{
 "cells": [
  {
   "cell_type": "code",
   "execution_count": 19,
   "metadata": {},
   "outputs": [],
   "source": [
    "import os\n",
    "import csv\n",
    "import sys\n",
    "import numpy as np\n",
    "from torch.utils.data import Dataset, DataLoader\n",
    "sys.path.insert(1, '../')\n",
    "from lxrt.SlowFast.slowfast.config.defaults import get_cfg"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 3,
   "metadata": {},
   "outputs": [],
   "source": [
    "from lxrt.SlowFast.slowfast.datasets.tgif_direct import TGIF"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 4,
   "metadata": {},
   "outputs": [],
   "source": [
    "cfg = get_cfg()\n",
    "cfg_file = \"../lxrt/SlowFast/configs/Kinetics/c2/SLOWFAST_8x8_R50.yaml\"\n",
    "cfg.merge_from_file(cfg_file)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 74,
   "metadata": {},
   "outputs": [],
   "source": [
    "\n",
    "class TGIFDataset(Dataset):\n",
    "    def __init__(self, dataset_name='train', data_type=None, dataframe_dir=None, vocab_dir=None):\n",
    "        self.dataframe_dir = dataframe_dir # of the form data/tgif/vocabulary\n",
    "        self.vocab_dir = vocab_dir # of the form data/tgif/dataframe\n",
    "        self.data_type = data_type # 'TRANS'\n",
    "        self.dataset_name = dataset_name # 'train' or 'val' or 'test'\n",
    "\n",
    "        self.csv = self.read_from_csvfile()\n",
    "        self.header2idx = self.header2idx()\n",
    "        self.gif_names = self.csv[:,self.header2idx['gif_name']]\n",
    "        self.gif_tensor = None\n",
    "        self.questions = self.csv[:,self.header2idx['question']]\n",
    "        self.answers = self.csv[:,self.header2idx['answer']]\n",
    "        self.mc_options = self.csv[:,self.header2idx['a1']:self.header2idx['a5']+1]\n",
    "        ## GIF LOADER ##\n",
    "        ## NOTE: May have to change the relative path of gif dir as \n",
    "        ## an extra argument to TGIF class init\n",
    "        loader  = TGIF(cfg, \"train\")\n",
    "        self.get_gif_tensor = loader.__getitem__\n",
    "        \n",
    "    def __getitem__(self, i): # whats the argument for this\n",
    "    \t#gif_path = os.path.join(self.dataframe_dir, 'gif_tensors')\n",
    "    \t#pick up ith gif_tensor\n",
    "        #NOTE: gif_path is only the gif name, not the relative path\n",
    "        # REturn value: tuple (slow frames, fast frames) where frame -> (t, 3, h, w)\n",
    "        self.gif_tensor = self.get_gif_tensor(self.gif_names[i])\n",
    "        return self.gif_tensor, self.questions[i], list(self.mc_options[i]), self.answers[i]\n",
    "    \n",
    "    def __len__(self):\n",
    "        return len(self.questions)\n",
    "\n",
    "    def header2idx(self):\n",
    "    \treturn {'gif_name':0,'question':1,'a1':2,'a2':3,'a3':4,'a4':5,'a5':6,'answer':7,'vid_id':8,'key':9}\n",
    "\n",
    "    def read_from_csvfile(self):\n",
    "        assert self.data_type in ['TRANS', 'ACTION'] # ACTION just for starting, will be using TRANS finally\n",
    "\n",
    "        self.total_q=[]\n",
    "        if self.data_type=='TRANS':\n",
    "            train_data_path = os.path.join(self.dataframe_dir, 'Train_transition_question.csv')\n",
    "            test_data_path = os.path.join(self.dataframe_dir, 'Test_transition_question.csv')\n",
    "\n",
    "        elif self.data_type=='ACTION':\n",
    "            train_data_path = os.path.join(self.dataframe_dir, 'Train_action_question.csv')\n",
    "            test_data_path = os.path.join(self.dataframe_dir, 'Test_action_question.csv')\n",
    "\n",
    "        if not (os.path.exists(train_data_path) and os.path.exists(test_data_path)):\n",
    "            print(\"Does not exist\")\n",
    "\n",
    "        csv_data=[]\n",
    "        if self.dataset_name=='train':\n",
    "        \twith open(train_data_path) as file:\n",
    "        \t\tcsv_reader = csv.reader(file, delimiter='\\t')\n",
    "        \t\tfor row in csv_reader:\n",
    "        \t\t\tcsv_data.append(row)\n",
    "        elif self.dataset_name=='test':\n",
    "        \twith open(test_data_path) as file:\n",
    "        \t\tcsv_reader = csv.reader(file, delimiter='\\t')\n",
    "        \t\tfor row in csv_reader:\n",
    "        \t\t\tcsv_data.append(row)\n",
    "        csv_data.pop(0)\n",
    "\n",
    "        return np.asarray(csv_data)"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 75,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nkgxdo19En1qf37ejo1_250.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n"
     ]
    },
    {
     "data": {
      "text/plain": [
       "([tensor([[[[-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
       "            [-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
       "            [-1.9999, -1.9999, -1.9990,  ..., -1.9994, -1.9998, -1.9998],\n",
       "            ...,\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000]],\n",
       "  \n",
       "           [[-1.9999, -1.9999, -1.9999,  ..., -1.9998, -1.9997, -1.9997],\n",
       "            [-1.9999, -1.9999, -1.9999,  ..., -1.9998, -1.9997, -1.9997],\n",
       "            [-1.9999, -1.9999, -1.9992,  ..., -1.9993, -1.9998, -1.9998],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9974, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9975, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9975, -1.9999, -1.9999]],\n",
       "  \n",
       "           [[-1.9999, -1.9999, -1.9999,  ..., -1.9999, -1.9999, -1.9999],\n",
       "            [-1.9999, -1.9999, -1.9999,  ..., -1.9999, -1.9999, -1.9999],\n",
       "            [-1.9999, -1.9999, -1.9992,  ..., -1.9995, -1.9999, -1.9999],\n",
       "            ...,\n",
       "            [-1.9998, -1.9998, -1.9995,  ..., -1.9976, -2.0000, -2.0000],\n",
       "            [-1.9997, -1.9997, -1.9995,  ..., -1.9976, -2.0000, -2.0000],\n",
       "            [-1.9998, -1.9998, -1.9995,  ..., -1.9974, -1.9998, -1.9998]],\n",
       "  \n",
       "           ...,\n",
       "  \n",
       "           [[-2.0000, -2.0000, -2.0000,  ..., -1.9991, -1.9988, -1.9988],\n",
       "            [-2.0000, -2.0000, -2.0000,  ..., -1.9991, -1.9988, -1.9988],\n",
       "            [-2.0000, -2.0000, -1.9993,  ..., -1.9988, -1.9987, -1.9987],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9973, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9973, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9973, -2.0000, -2.0000]],\n",
       "  \n",
       "           [[-1.9998, -1.9998, -1.9998,  ..., -2.0000, -2.0000, -2.0000],\n",
       "            [-1.9998, -1.9998, -1.9998,  ..., -2.0000, -2.0000, -2.0000],\n",
       "            [-1.9997, -1.9997, -1.9997,  ..., -1.9995, -1.9999, -1.9999],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9964, -1.9996, -1.9996],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9964, -1.9997, -1.9997],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9965, -1.9997, -1.9997]],\n",
       "  \n",
       "           [[-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9990, -1.9997, -1.9997],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9998, -1.9998],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9997, -1.9997]]],\n",
       "  \n",
       "  \n",
       "          [[[-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
       "            [-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
       "            [-1.9998, -1.9998, -1.9990,  ..., -1.9986, -1.9997, -1.9997],\n",
       "            ...,\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9965, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9966, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9967, -2.0000, -2.0000]],\n",
       "  \n",
       "           [[-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9992,  ..., -1.9990, -1.9999, -1.9999],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9965, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9966, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9966, -1.9998, -1.9998]],\n",
       "  \n",
       "           [[-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9999, -1.9999],\n",
       "            [-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9999, -1.9999],\n",
       "            [-2.0000, -2.0000, -1.9993,  ..., -1.9990, -1.9998, -1.9998],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9996,  ..., -1.9966, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -1.9999, -1.9999]],\n",
       "  \n",
       "           ...,\n",
       "  \n",
       "           [[-1.9999, -1.9999, -1.9999,  ..., -1.9996, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9999,  ..., -1.9996, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9992,  ..., -1.9991, -1.9999, -1.9999],\n",
       "            ...,\n",
       "            [-1.9999, -1.9999, -1.9996,  ..., -1.9963, -1.9997, -1.9997],\n",
       "            [-1.9999, -1.9999, -1.9996,  ..., -1.9964, -1.9997, -1.9997],\n",
       "            [-1.9999, -1.9999, -1.9996,  ..., -1.9965, -1.9998, -1.9998]],\n",
       "  \n",
       "           [[-1.9995, -1.9995, -1.9996,  ..., -1.9995, -1.9999, -1.9999],\n",
       "            [-1.9995, -1.9995, -1.9996,  ..., -1.9995, -1.9999, -1.9999],\n",
       "            [-1.9994, -1.9994, -1.9995,  ..., -1.9989, -1.9997, -1.9997],\n",
       "            ...,\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9960, -1.9994, -1.9994],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9960, -1.9994, -1.9994],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9961, -1.9994, -1.9994]],\n",
       "  \n",
       "           [[-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
       "            [-1.9999, -1.9999, -1.9996,  ..., -1.9987, -1.9996, -1.9996],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9997, -1.9997],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9996, -1.9996]]],\n",
       "  \n",
       "  \n",
       "          [[[-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
       "            [-1.9998, -1.9998, -1.9989,  ..., -1.9987, -1.9999, -1.9999],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9964, -1.9997, -1.9997],\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9965, -1.9997, -1.9997],\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -1.9998, -1.9998]],\n",
       "  \n",
       "           [[-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9991,  ..., -1.9988, -1.9998, -1.9998],\n",
       "            ...,\n",
       "            [-1.9995, -1.9995, -1.9992,  ..., -1.9965, -2.0000, -2.0000],\n",
       "            [-1.9994, -1.9994, -1.9991,  ..., -1.9966, -2.0000, -2.0000],\n",
       "            [-1.9996, -1.9996, -1.9993,  ..., -1.9966, -1.9999, -1.9999]],\n",
       "  \n",
       "           [[-2.0000, -2.0000, -2.0000,  ..., -1.9997, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -2.0000,  ..., -1.9997, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9992,  ..., -1.9989, -1.9999, -1.9999],\n",
       "            ...,\n",
       "            [-1.9999, -1.9999, -1.9996,  ..., -1.9966, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9996,  ..., -1.9966, -1.9999, -1.9999]],\n",
       "  \n",
       "           ...,\n",
       "  \n",
       "           [[-1.9997, -1.9997, -1.9997,  ..., -1.9992, -1.9997, -1.9997],\n",
       "            [-1.9997, -1.9997, -1.9997,  ..., -1.9992, -1.9997, -1.9997],\n",
       "            [-1.9996, -1.9996, -1.9990,  ..., -1.9987, -1.9995, -1.9995],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9996,  ..., -1.9960, -1.9992, -1.9992],\n",
       "            [-2.0000, -2.0000, -1.9996,  ..., -1.9960, -1.9993, -1.9993],\n",
       "            [-2.0000, -2.0000, -1.9996,  ..., -1.9960, -1.9993, -1.9993]],\n",
       "  \n",
       "           [[-1.9997, -1.9997, -1.9998,  ..., -1.9992, -2.0000, -2.0000],\n",
       "            [-1.9997, -1.9997, -1.9998,  ..., -1.9992, -2.0000, -2.0000],\n",
       "            [-1.9996, -1.9996, -1.9997,  ..., -1.9986, -2.0000, -2.0000],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9957, -1.9992, -1.9992],\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9957, -1.9992, -1.9992],\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9958, -1.9992, -1.9992]],\n",
       "  \n",
       "           [[-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
       "            [-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
       "            [-1.9998, -1.9998, -1.9995,  ..., -1.9983, -1.9992, -1.9992],\n",
       "            ...,\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9995, -1.9995],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994]]]]),\n",
       "  tensor([[[[-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
       "            [-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
       "            [-1.9999, -1.9999, -1.9990,  ..., -1.9994, -1.9998, -1.9998],\n",
       "            ...,\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000]],\n",
       "  \n",
       "           [[-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
       "            [-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
       "            [-1.9999, -1.9999, -1.9990,  ..., -1.9994, -1.9998, -1.9998],\n",
       "            ...,\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000]],\n",
       "  \n",
       "           [[-2.0000, -2.0000, -2.0000,  ..., -1.9999, -1.9999, -1.9999],\n",
       "            [-2.0000, -2.0000, -2.0000,  ..., -1.9999, -1.9999, -1.9999],\n",
       "            [-2.0000, -2.0000, -1.9991,  ..., -1.9995, -1.9999, -1.9999],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9973, -1.9998, -1.9998],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9973, -1.9998, -1.9998],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9973, -1.9997, -1.9997]],\n",
       "  \n",
       "           ...,\n",
       "  \n",
       "           [[-1.9998, -1.9998, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
       "            [-1.9998, -1.9998, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
       "            [-1.9998, -1.9998, -1.9998,  ..., -1.9991, -1.9999, -1.9999],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000]],\n",
       "  \n",
       "           [[-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9990, -1.9997, -1.9997],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9998, -1.9998],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9997, -1.9997]],\n",
       "  \n",
       "           [[-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9990, -1.9997, -1.9997],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9998, -1.9998],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9997, -1.9997]]],\n",
       "  \n",
       "  \n",
       "          [[[-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
       "            [-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
       "            [-1.9998, -1.9998, -1.9990,  ..., -1.9986, -1.9997, -1.9997],\n",
       "            ...,\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9965, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9966, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9967, -2.0000, -2.0000]],\n",
       "  \n",
       "           [[-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
       "            [-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
       "            [-1.9998, -1.9998, -1.9990,  ..., -1.9986, -1.9997, -1.9997],\n",
       "            ...,\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9965, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9966, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9967, -2.0000, -2.0000]],\n",
       "  \n",
       "           [[-1.9999, -1.9999, -1.9999,  ..., -1.9999, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9999,  ..., -1.9999, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9990,  ..., -1.9990, -2.0000, -2.0000],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9965, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9966, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9966, -1.9998, -1.9998]],\n",
       "  \n",
       "           ...,\n",
       "  \n",
       "           [[-1.9995, -1.9995, -1.9996,  ..., -1.9994, -1.9999, -1.9999],\n",
       "            [-1.9995, -1.9995, -1.9996,  ..., -1.9994, -1.9999, -1.9999],\n",
       "            [-1.9996, -1.9996, -1.9996,  ..., -1.9988, -1.9997, -1.9997],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000]],\n",
       "  \n",
       "           [[-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
       "            [-1.9999, -1.9999, -1.9996,  ..., -1.9987, -1.9996, -1.9996],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9997, -1.9997],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9996, -1.9996]],\n",
       "  \n",
       "           [[-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
       "            [-1.9999, -1.9999, -1.9996,  ..., -1.9987, -1.9996, -1.9996],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9997, -1.9997],\n",
       "            [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9996, -1.9996]]],\n",
       "  \n",
       "  \n",
       "          [[[-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
       "            [-1.9998, -1.9998, -1.9989,  ..., -1.9987, -1.9999, -1.9999],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9964, -1.9997, -1.9997],\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9965, -1.9997, -1.9997],\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -1.9998, -1.9998]],\n",
       "  \n",
       "           [[-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
       "            [-1.9998, -1.9998, -1.9989,  ..., -1.9987, -1.9999, -1.9999],\n",
       "            ...,\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9964, -1.9997, -1.9997],\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9965, -1.9997, -1.9997],\n",
       "            [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -1.9998, -1.9998]],\n",
       "  \n",
       "           [[-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
       "            [-2.0000, -2.0000, -1.9990,  ..., -1.9989, -1.9999, -1.9999],\n",
       "            ...,\n",
       "            [-1.9998, -1.9998, -1.9996,  ..., -1.9965, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9996,  ..., -1.9966, -2.0000, -2.0000],\n",
       "            [-1.9999, -1.9999, -1.9996,  ..., -1.9966, -1.9998, -1.9998]],\n",
       "  \n",
       "           ...,\n",
       "  \n",
       "           [[-1.9993, -1.9993, -1.9994,  ..., -1.9989, -1.9995, -1.9995],\n",
       "            [-1.9993, -1.9993, -1.9994,  ..., -1.9989, -1.9995, -1.9995],\n",
       "            [-1.9994, -1.9994, -1.9994,  ..., -1.9983, -1.9993, -1.9993],\n",
       "            ...,\n",
       "            [-1.9997, -1.9997, -1.9997,  ..., -1.9959, -1.9997, -1.9997],\n",
       "            [-1.9997, -1.9997, -1.9997,  ..., -1.9959, -1.9997, -1.9997],\n",
       "            [-1.9997, -1.9997, -1.9997,  ..., -1.9959, -1.9997, -1.9997]],\n",
       "  \n",
       "           [[-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
       "            [-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
       "            [-1.9998, -1.9998, -1.9995,  ..., -1.9983, -1.9992, -1.9992],\n",
       "            ...,\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9995, -1.9995],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994]],\n",
       "  \n",
       "           [[-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
       "            [-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
       "            [-1.9998, -1.9998, -1.9995,  ..., -1.9983, -1.9992, -1.9992],\n",
       "            ...,\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9995, -1.9995],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994],\n",
       "            [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994]]]])],\n",
       " 'What does the woman do before places hands in lap ?',\n",
       " [\"stoke man's face\",\n",
       "  'wink',\n",
       "  'chew food',\n",
       "  'clap the steer wheel',\n",
       "  'has hands on face'],\n",
       " '4')"
      ]
     },
     "execution_count": 75,
     "metadata": {},
     "output_type": "execute_result"
    }
   ],
   "source": [
    "data_file = \"../../../../../../IDL/project/tgif-qa/dataset/\"\n",
    "dataset = TGIFDataset(dataset_name='train', data_type='TRANS', dataframe_dir=data_file, vocab_dir=None)\n",
    "dataset[0]"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": 76,
   "metadata": {
    "scrolled": true
   },
   "outputs": [
    {
     "name": "stdout",
     "output_type": "stream",
     "text": [
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nkgxdo19En1qf37ejo1_250.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nkgxdo19En1qf37ejo1_250.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nqc2mbmU2J1uxhtnwo1_400.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nqc2mbmU2J1uxhtnwo1_400.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nesl84pTnp1tmddexo1_400.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nesl84pTnp1tmddexo1_400.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_ncvuw1IZCS1sma80lo1_400.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_ncvuw1IZCS1sma80lo1_400.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nkgx9nVFar1qf37ejo1_250.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nkgx9nVFar1qf37ejo1_250.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_meiblncAdv1regdl0o1_r1_500.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_meiblncAdv1regdl0o1_r1_500.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nknb86YzIa1u3knpvo5_400.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nknb86YzIa1u3knpvo5_400.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_n8xk3uXXdI1qh67ano1_400.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_n8xk3uXXdI1qh67ano1_400.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_ngx732ghCk1sd2bbjo1_500.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_ngx732ghCk1sd2bbjo1_500.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nekvrnW3rh1sxaoaro1_400.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nekvrnW3rh1sxaoaro1_400.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_ncvyrtwT5G1roqomqo1_400.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_ncvyrtwT5G1roqomqo1_400.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nbnhln247i1tjdttgo1_250.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nbnhln247i1tjdttgo1_250.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nk172bbdPI1u1lr18o1_250.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nk172bbdPI1u1lr18o1_250.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_n9ozi8OUvm1sibzdqo3_250.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_n9ozi8OUvm1sibzdqo3_250.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nf4li6QwKK1sp0mkho1_400.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nf4li6QwKK1sp0mkho1_400.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nke4tcNUsU1trtmm1o1_250.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "/users/cdwivedi/RL_EXP/IDL/project/tgif-qa/code/dataset/tgif/gifs/tumblr_nke4tcNUsU1trtmm1o1_250.gi/*\n",
      "torch.Size([3, 8, 256, 256]) torch.Size([3, 32, 256, 256])\n",
      "tensor:  [tensor([[[[[-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
      "           [-1.9999, -1.9999, -1.9990,  ..., -1.9994, -1.9998, -1.9998],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9999,  ..., -1.9998, -1.9997, -1.9997],\n",
      "           [-1.9999, -1.9999, -1.9999,  ..., -1.9998, -1.9997, -1.9997],\n",
      "           [-1.9999, -1.9999, -1.9992,  ..., -1.9993, -1.9998, -1.9998],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9974, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9975, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9975, -1.9999, -1.9999]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9999,  ..., -1.9999, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9999,  ..., -1.9999, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9992,  ..., -1.9995, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-1.9998, -1.9998, -1.9995,  ..., -1.9976, -2.0000, -2.0000],\n",
      "           [-1.9997, -1.9997, -1.9995,  ..., -1.9976, -2.0000, -2.0000],\n",
      "           [-1.9998, -1.9998, -1.9995,  ..., -1.9974, -1.9998, -1.9998]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9991, -1.9988, -1.9988],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9991, -1.9988, -1.9988],\n",
      "           [-2.0000, -2.0000, -1.9993,  ..., -1.9988, -1.9987, -1.9987],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9973, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9973, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9973, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-1.9998, -1.9998, -1.9998,  ..., -2.0000, -2.0000, -2.0000],\n",
      "           [-1.9998, -1.9998, -1.9998,  ..., -2.0000, -2.0000, -2.0000],\n",
      "           [-1.9997, -1.9997, -1.9997,  ..., -1.9995, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9964, -1.9996, -1.9996],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9964, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9965, -1.9997, -1.9997]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9990, -1.9997, -1.9997],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9997, -1.9997]]],\n",
      "\n",
      "\n",
      "         [[[-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
      "           [-1.9998, -1.9998, -1.9990,  ..., -1.9986, -1.9997, -1.9997],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9965, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9967, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9992,  ..., -1.9990, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9965, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9966, -1.9998, -1.9998]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9999, -1.9999],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9999, -1.9999],\n",
      "           [-2.0000, -2.0000, -1.9993,  ..., -1.9990, -1.9998, -1.9998],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9996,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -1.9999, -1.9999]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9999,  ..., -1.9996, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9999,  ..., -1.9996, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9992,  ..., -1.9991, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9963, -1.9997, -1.9997],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9964, -1.9997, -1.9997],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9965, -1.9998, -1.9998]],\n",
      "\n",
      "          [[-1.9995, -1.9995, -1.9996,  ..., -1.9995, -1.9999, -1.9999],\n",
      "           [-1.9995, -1.9995, -1.9996,  ..., -1.9995, -1.9999, -1.9999],\n",
      "           [-1.9994, -1.9994, -1.9995,  ..., -1.9989, -1.9997, -1.9997],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9960, -1.9994, -1.9994],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9960, -1.9994, -1.9994],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9961, -1.9994, -1.9994]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9987, -1.9996, -1.9996],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9996, -1.9996]]],\n",
      "\n",
      "\n",
      "         [[[-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
      "           [-1.9998, -1.9998, -1.9989,  ..., -1.9987, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9964, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9965, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -1.9998, -1.9998]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9991,  ..., -1.9988, -1.9998, -1.9998],\n",
      "           ...,\n",
      "           [-1.9995, -1.9995, -1.9992,  ..., -1.9965, -2.0000, -2.0000],\n",
      "           [-1.9994, -1.9994, -1.9991,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-1.9996, -1.9996, -1.9993,  ..., -1.9966, -1.9999, -1.9999]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9992,  ..., -1.9989, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9996,  ..., -1.9966, -1.9999, -1.9999]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9997, -1.9997, -1.9997,  ..., -1.9992, -1.9997, -1.9997],\n",
      "           [-1.9997, -1.9997, -1.9997,  ..., -1.9992, -1.9997, -1.9997],\n",
      "           [-1.9996, -1.9996, -1.9990,  ..., -1.9987, -1.9995, -1.9995],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9996,  ..., -1.9960, -1.9992, -1.9992],\n",
      "           [-2.0000, -2.0000, -1.9996,  ..., -1.9960, -1.9993, -1.9993],\n",
      "           [-2.0000, -2.0000, -1.9996,  ..., -1.9960, -1.9993, -1.9993]],\n",
      "\n",
      "          [[-1.9997, -1.9997, -1.9998,  ..., -1.9992, -2.0000, -2.0000],\n",
      "           [-1.9997, -1.9997, -1.9998,  ..., -1.9992, -2.0000, -2.0000],\n",
      "           [-1.9996, -1.9996, -1.9997,  ..., -1.9986, -2.0000, -2.0000],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9957, -1.9992, -1.9992],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9957, -1.9992, -1.9992],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9958, -1.9992, -1.9992]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
      "           [-1.9998, -1.9998, -1.9995,  ..., -1.9983, -1.9992, -1.9992],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9995, -1.9995],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994]]]],\n",
      "\n",
      "\n",
      "\n",
      "        [[[[-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
      "           [-1.9999, -1.9999, -1.9990,  ..., -1.9994, -1.9998, -1.9998],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9999,  ..., -1.9998, -1.9997, -1.9997],\n",
      "           [-1.9999, -1.9999, -1.9999,  ..., -1.9998, -1.9997, -1.9997],\n",
      "           [-1.9999, -1.9999, -1.9992,  ..., -1.9993, -1.9998, -1.9998],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9974, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9975, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9975, -1.9999, -1.9999]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9999,  ..., -1.9999, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9999,  ..., -1.9999, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9992,  ..., -1.9995, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-1.9998, -1.9998, -1.9995,  ..., -1.9976, -2.0000, -2.0000],\n",
      "           [-1.9997, -1.9997, -1.9995,  ..., -1.9976, -2.0000, -2.0000],\n",
      "           [-1.9998, -1.9998, -1.9995,  ..., -1.9974, -1.9998, -1.9998]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9991, -1.9988, -1.9988],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9991, -1.9988, -1.9988],\n",
      "           [-2.0000, -2.0000, -1.9993,  ..., -1.9988, -1.9987, -1.9987],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9973, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9973, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9973, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-1.9998, -1.9998, -1.9998,  ..., -2.0000, -2.0000, -2.0000],\n",
      "           [-1.9998, -1.9998, -1.9998,  ..., -2.0000, -2.0000, -2.0000],\n",
      "           [-1.9997, -1.9997, -1.9997,  ..., -1.9995, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9964, -1.9996, -1.9996],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9964, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9965, -1.9997, -1.9997]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9990, -1.9997, -1.9997],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9997, -1.9997]]],\n",
      "\n",
      "\n",
      "         [[[-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
      "           [-1.9998, -1.9998, -1.9990,  ..., -1.9986, -1.9997, -1.9997],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9965, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9967, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9992,  ..., -1.9990, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9965, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9966, -1.9998, -1.9998]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9999, -1.9999],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9999, -1.9999],\n",
      "           [-2.0000, -2.0000, -1.9993,  ..., -1.9990, -1.9998, -1.9998],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9996,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -1.9999, -1.9999]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9999,  ..., -1.9996, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9999,  ..., -1.9996, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9992,  ..., -1.9991, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9963, -1.9997, -1.9997],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9964, -1.9997, -1.9997],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9965, -1.9998, -1.9998]],\n",
      "\n",
      "          [[-1.9995, -1.9995, -1.9996,  ..., -1.9995, -1.9999, -1.9999],\n",
      "           [-1.9995, -1.9995, -1.9996,  ..., -1.9995, -1.9999, -1.9999],\n",
      "           [-1.9994, -1.9994, -1.9995,  ..., -1.9989, -1.9997, -1.9997],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9960, -1.9994, -1.9994],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9960, -1.9994, -1.9994],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9961, -1.9994, -1.9994]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9987, -1.9996, -1.9996],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9996, -1.9996]]],\n",
      "\n",
      "\n",
      "         [[[-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
      "           [-1.9998, -1.9998, -1.9989,  ..., -1.9987, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9964, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9965, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -1.9998, -1.9998]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9991,  ..., -1.9988, -1.9998, -1.9998],\n",
      "           ...,\n",
      "           [-1.9995, -1.9995, -1.9992,  ..., -1.9965, -2.0000, -2.0000],\n",
      "           [-1.9994, -1.9994, -1.9991,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-1.9996, -1.9996, -1.9993,  ..., -1.9966, -1.9999, -1.9999]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9992,  ..., -1.9989, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9996,  ..., -1.9966, -1.9999, -1.9999]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9997, -1.9997, -1.9997,  ..., -1.9992, -1.9997, -1.9997],\n",
      "           [-1.9997, -1.9997, -1.9997,  ..., -1.9992, -1.9997, -1.9997],\n",
      "           [-1.9996, -1.9996, -1.9990,  ..., -1.9987, -1.9995, -1.9995],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9996,  ..., -1.9960, -1.9992, -1.9992],\n",
      "           [-2.0000, -2.0000, -1.9996,  ..., -1.9960, -1.9993, -1.9993],\n",
      "           [-2.0000, -2.0000, -1.9996,  ..., -1.9960, -1.9993, -1.9993]],\n",
      "\n",
      "          [[-1.9997, -1.9997, -1.9998,  ..., -1.9992, -2.0000, -2.0000],\n",
      "           [-1.9997, -1.9997, -1.9998,  ..., -1.9992, -2.0000, -2.0000],\n",
      "           [-1.9996, -1.9996, -1.9997,  ..., -1.9986, -2.0000, -2.0000],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9957, -1.9992, -1.9992],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9957, -1.9992, -1.9992],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9958, -1.9992, -1.9992]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
      "           [-1.9998, -1.9998, -1.9995,  ..., -1.9983, -1.9992, -1.9992],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9995, -1.9995],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994]]]],\n",
      "\n",
      "\n",
      "\n",
      "        [[[[-1.9914, -1.9918, -1.9923,  ..., -1.9882, -1.9883, -1.9884],\n",
      "           [-1.9917, -1.9920, -1.9924,  ..., -1.9884, -1.9885, -1.9886],\n",
      "           [-1.9919, -1.9921, -1.9924,  ..., -1.9886, -1.9888, -1.9889],\n",
      "           ...,\n",
      "           [-1.9930, -1.9930, -1.9930,  ..., -1.9926, -1.9927, -1.9926],\n",
      "           [-1.9932, -1.9933, -1.9934,  ..., -1.9925, -1.9927, -1.9926],\n",
      "           [-1.9935, -1.9936, -1.9937,  ..., -1.9926, -1.9928, -1.9926]],\n",
      "\n",
      "          [[-1.9917, -1.9919, -1.9923,  ..., -1.9883, -1.9883, -1.9882],\n",
      "           [-1.9919, -1.9920, -1.9922,  ..., -1.9887, -1.9886, -1.9886],\n",
      "           [-1.9919, -1.9919, -1.9922,  ..., -1.9889, -1.9890, -1.9890],\n",
      "           ...,\n",
      "           [-1.9938, -1.9936, -1.9935,  ..., -1.9862, -1.9854, -1.9848],\n",
      "           [-1.9945, -1.9942, -1.9941,  ..., -1.9865, -1.9857, -1.9850],\n",
      "           [-1.9946, -1.9944, -1.9942,  ..., -1.9869, -1.9861, -1.9852]],\n",
      "\n",
      "          [[-1.9915, -1.9916, -1.9919,  ..., -1.9882, -1.9882, -1.9882],\n",
      "           [-1.9915, -1.9917, -1.9919,  ..., -1.9885, -1.9886, -1.9886],\n",
      "           [-1.9917, -1.9918, -1.9920,  ..., -1.9887, -1.9888, -1.9888],\n",
      "           ...,\n",
      "           [-1.9875, -1.9879, -1.9885,  ..., -1.9843, -1.9843, -1.9843],\n",
      "           [-1.9869, -1.9877, -1.9887,  ..., -1.9843, -1.9843, -1.9843],\n",
      "           [-1.9863, -1.9875, -1.9890,  ..., -1.9843, -1.9843, -1.9843]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9918, -1.9919, -1.9920,  ..., -1.9887, -1.9887, -1.9887],\n",
      "           [-1.9918, -1.9919, -1.9920,  ..., -1.9886, -1.9886, -1.9886],\n",
      "           [-1.9918, -1.9919, -1.9920,  ..., -1.9886, -1.9886, -1.9886],\n",
      "           ...,\n",
      "           [-1.9960, -1.9960, -1.9960,  ..., -1.9913, -1.9913, -1.9913],\n",
      "           [-1.9960, -1.9960, -1.9960,  ..., -1.9903, -1.9903, -1.9903],\n",
      "           [-1.9960, -1.9960, -1.9960,  ..., -1.9895, -1.9895, -1.9895]],\n",
      "\n",
      "          [[-1.9918, -1.9919, -1.9920,  ..., -1.9886, -1.9886, -1.9886],\n",
      "           [-1.9918, -1.9919, -1.9920,  ..., -1.9885, -1.9885, -1.9885],\n",
      "           [-1.9918, -1.9919, -1.9920,  ..., -1.9885, -1.9885, -1.9885],\n",
      "           ...,\n",
      "           [-1.9960, -1.9960, -1.9960,  ..., -1.9914, -1.9914, -1.9914],\n",
      "           [-1.9960, -1.9960, -1.9960,  ..., -1.9904, -1.9904, -1.9904],\n",
      "           [-1.9960, -1.9960, -1.9960,  ..., -1.9896, -1.9896, -1.9896]],\n",
      "\n",
      "          [[-1.9918, -1.9919, -1.9920,  ..., -1.9886, -1.9886, -1.9886],\n",
      "           [-1.9918, -1.9919, -1.9920,  ..., -1.9885, -1.9885, -1.9885],\n",
      "           [-1.9918, -1.9919, -1.9920,  ..., -1.9885, -1.9885, -1.9885],\n",
      "           ...,\n",
      "           [-1.9947, -1.9937, -1.9925,  ..., -1.9914, -1.9914, -1.9914],\n",
      "           [-1.9947, -1.9937, -1.9925,  ..., -1.9904, -1.9904, -1.9904],\n",
      "           [-1.9947, -1.9937, -1.9925,  ..., -1.9896, -1.9896, -1.9896]]],\n",
      "\n",
      "\n",
      "         [[[-1.9919, -1.9924, -1.9929,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9921, -1.9925, -1.9930,  ..., -1.9894, -1.9895, -1.9895],\n",
      "           [-1.9923, -1.9926, -1.9930,  ..., -1.9898, -1.9899, -1.9899],\n",
      "           ...,\n",
      "           [-1.9945, -1.9945, -1.9946,  ..., -1.9937, -1.9939, -1.9938],\n",
      "           [-1.9949, -1.9949, -1.9950,  ..., -1.9934, -1.9936, -1.9935],\n",
      "           [-1.9950, -1.9951, -1.9952,  ..., -1.9931, -1.9933, -1.9931]],\n",
      "\n",
      "          [[-1.9927, -1.9927, -1.9930,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9926, -1.9925, -1.9928,  ..., -1.9894, -1.9894, -1.9894],\n",
      "           [-1.9926, -1.9925, -1.9927,  ..., -1.9898, -1.9898, -1.9898],\n",
      "           ...,\n",
      "           [-1.9944, -1.9945, -1.9946,  ..., -1.9870, -1.9860, -1.9850],\n",
      "           [-1.9953, -1.9953, -1.9953,  ..., -1.9872, -1.9863, -1.9854],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9875, -1.9869, -1.9863]],\n",
      "\n",
      "          [[-1.9921, -1.9922, -1.9926,  ..., -1.9890, -1.9890, -1.9889],\n",
      "           [-1.9922, -1.9923, -1.9926,  ..., -1.9894, -1.9894, -1.9894],\n",
      "           [-1.9923, -1.9924, -1.9927,  ..., -1.9898, -1.9898, -1.9898],\n",
      "           ...,\n",
      "           [-1.9880, -1.9883, -1.9890,  ..., -1.9850, -1.9850, -1.9850],\n",
      "           [-1.9874, -1.9882, -1.9893,  ..., -1.9850, -1.9850, -1.9850],\n",
      "           [-1.9870, -1.9881, -1.9896,  ..., -1.9850, -1.9850, -1.9850]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9928, -1.9928, -1.9929,  ..., -1.9891, -1.9891, -1.9891],\n",
      "           [-1.9928, -1.9928, -1.9929,  ..., -1.9893, -1.9893, -1.9893],\n",
      "           [-1.9928, -1.9928, -1.9929,  ..., -1.9895, -1.9895, -1.9895],\n",
      "           ...,\n",
      "           [-1.9969, -1.9969, -1.9969,  ..., -1.9925, -1.9925, -1.9925],\n",
      "           [-1.9969, -1.9969, -1.9969,  ..., -1.9915, -1.9915, -1.9915],\n",
      "           [-1.9969, -1.9969, -1.9969,  ..., -1.9907, -1.9907, -1.9907]],\n",
      "\n",
      "          [[-1.9928, -1.9928, -1.9929,  ..., -1.9891, -1.9891, -1.9891],\n",
      "           [-1.9928, -1.9928, -1.9929,  ..., -1.9892, -1.9892, -1.9892],\n",
      "           [-1.9928, -1.9928, -1.9929,  ..., -1.9894, -1.9894, -1.9894],\n",
      "           ...,\n",
      "           [-1.9970, -1.9970, -1.9970,  ..., -1.9925, -1.9925, -1.9925],\n",
      "           [-1.9970, -1.9970, -1.9970,  ..., -1.9915, -1.9915, -1.9915],\n",
      "           [-1.9970, -1.9970, -1.9970,  ..., -1.9907, -1.9907, -1.9907]],\n",
      "\n",
      "          [[-1.9928, -1.9928, -1.9929,  ..., -1.9890, -1.9890, -1.9890],\n",
      "           [-1.9928, -1.9928, -1.9929,  ..., -1.9891, -1.9891, -1.9891],\n",
      "           [-1.9928, -1.9928, -1.9929,  ..., -1.9893, -1.9893, -1.9893],\n",
      "           ...,\n",
      "           [-1.9953, -1.9943, -1.9931,  ..., -1.9925, -1.9925, -1.9925],\n",
      "           [-1.9953, -1.9943, -1.9931,  ..., -1.9915, -1.9915, -1.9915],\n",
      "           [-1.9953, -1.9943, -1.9931,  ..., -1.9907, -1.9907, -1.9907]]],\n",
      "\n",
      "\n",
      "         [[[-1.9930, -1.9934, -1.9940,  ..., -1.9897, -1.9898, -1.9900],\n",
      "           [-1.9932, -1.9936, -1.9940,  ..., -1.9903, -1.9905, -1.9905],\n",
      "           [-1.9934, -1.9937, -1.9940,  ..., -1.9909, -1.9910, -1.9911],\n",
      "           ...,\n",
      "           [-1.9954, -1.9954, -1.9954,  ..., -1.9948, -1.9949, -1.9948],\n",
      "           [-1.9957, -1.9958, -1.9959,  ..., -1.9946, -1.9948, -1.9947],\n",
      "           [-1.9960, -1.9961, -1.9962,  ..., -1.9944, -1.9946, -1.9945]],\n",
      "\n",
      "          [[-1.9930, -1.9932, -1.9936,  ..., -1.9898, -1.9898, -1.9899],\n",
      "           [-1.9932, -1.9933, -1.9936,  ..., -1.9904, -1.9905, -1.9905],\n",
      "           [-1.9932, -1.9933, -1.9935,  ..., -1.9910, -1.9911, -1.9911],\n",
      "           ...,\n",
      "           [-1.9952, -1.9953, -1.9954,  ..., -1.9881, -1.9870, -1.9860],\n",
      "           [-1.9961, -1.9961, -1.9961,  ..., -1.9882, -1.9873, -1.9864],\n",
      "           [-1.9965, -1.9965, -1.9964,  ..., -1.9888, -1.9881, -1.9873]],\n",
      "\n",
      "          [[-1.9931, -1.9933, -1.9936,  ..., -1.9898, -1.9898, -1.9897],\n",
      "           [-1.9932, -1.9933, -1.9936,  ..., -1.9904, -1.9904, -1.9904],\n",
      "           [-1.9933, -1.9935, -1.9937,  ..., -1.9909, -1.9910, -1.9910],\n",
      "           ...,\n",
      "           [-1.9885, -1.9888, -1.9895,  ..., -1.9861, -1.9861, -1.9861],\n",
      "           [-1.9880, -1.9888, -1.9899,  ..., -1.9861, -1.9861, -1.9861],\n",
      "           [-1.9876, -1.9888, -1.9902,  ..., -1.9861, -1.9861, -1.9861]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9936, -1.9937, -1.9938,  ..., -1.9910, -1.9910, -1.9910],\n",
      "           [-1.9936, -1.9937, -1.9938,  ..., -1.9911, -1.9911, -1.9911],\n",
      "           [-1.9936, -1.9937, -1.9938,  ..., -1.9912, -1.9912, -1.9912],\n",
      "           ...,\n",
      "           [-1.9975, -1.9975, -1.9975,  ..., -1.9936, -1.9936, -1.9936],\n",
      "           [-1.9975, -1.9975, -1.9975,  ..., -1.9926, -1.9926, -1.9926],\n",
      "           [-1.9975, -1.9975, -1.9975,  ..., -1.9918, -1.9918, -1.9918]],\n",
      "\n",
      "          [[-1.9936, -1.9937, -1.9938,  ..., -1.9910, -1.9910, -1.9910],\n",
      "           [-1.9936, -1.9937, -1.9938,  ..., -1.9911, -1.9911, -1.9911],\n",
      "           [-1.9936, -1.9937, -1.9938,  ..., -1.9912, -1.9912, -1.9912],\n",
      "           ...,\n",
      "           [-1.9976, -1.9976, -1.9976,  ..., -1.9935, -1.9935, -1.9935],\n",
      "           [-1.9976, -1.9976, -1.9976,  ..., -1.9925, -1.9925, -1.9925],\n",
      "           [-1.9976, -1.9976, -1.9976,  ..., -1.9917, -1.9917, -1.9917]],\n",
      "\n",
      "          [[-1.9936, -1.9937, -1.9938,  ..., -1.9908, -1.9908, -1.9908],\n",
      "           [-1.9936, -1.9937, -1.9938,  ..., -1.9909, -1.9909, -1.9909],\n",
      "           [-1.9936, -1.9937, -1.9938,  ..., -1.9910, -1.9910, -1.9910],\n",
      "           ...,\n",
      "           [-1.9956, -1.9947, -1.9934,  ..., -1.9935, -1.9935, -1.9935],\n",
      "           [-1.9956, -1.9947, -1.9934,  ..., -1.9925, -1.9925, -1.9925],\n",
      "           [-1.9956, -1.9947, -1.9934,  ..., -1.9917, -1.9917, -1.9917]]]],\n",
      "\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "\n",
      "        [[[[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9995, -1.9995, -1.9995,  ..., -1.9845, -1.9845, -1.9844],\n",
      "           [-1.9995, -1.9995, -1.9995,  ..., -1.9845, -1.9845, -1.9844],\n",
      "           [-1.9995, -1.9995, -1.9995,  ..., -1.9844, -1.9844, -1.9845]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9991, -1.9991, -1.9990,  ..., -1.9842, -1.9844, -1.9844],\n",
      "           [-1.9989, -1.9989, -1.9989,  ..., -1.9840, -1.9841, -1.9843],\n",
      "           [-1.9989, -1.9989, -1.9989,  ..., -1.9840, -1.9839, -1.9841]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9992, -1.9992, -1.9992,  ..., -1.9827, -1.9827, -1.9827],\n",
      "           [-1.9992, -1.9992, -1.9992,  ..., -1.9827, -1.9827, -1.9827],\n",
      "           [-1.9992, -1.9992, -1.9992,  ..., -1.9827, -1.9827, -1.9827]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9977, -1.9976, -1.9975,  ..., -1.9870, -1.9875, -1.9878],\n",
      "           [-1.9977, -1.9976, -1.9975,  ..., -1.9868, -1.9873, -1.9876],\n",
      "           [-1.9978, -1.9977, -1.9976,  ..., -1.9867, -1.9872, -1.9875]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9959, -1.9959, -1.9958,  ..., -1.9860, -1.9866, -1.9869],\n",
      "           [-1.9948, -1.9947, -1.9946,  ..., -1.9858, -1.9857, -1.9855],\n",
      "           [-1.9938, -1.9936, -1.9934,  ..., -1.9856, -1.9852, -1.9847]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9878, -1.9863, -1.9873],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9879, -1.9877, -1.9893],\n",
      "           [-1.9957, -1.9957, -1.9957,  ..., -1.9889, -1.9895, -1.9912]]],\n",
      "\n",
      "\n",
      "         [[[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9993, -1.9993, -1.9993,  ..., -1.9857, -1.9856, -1.9856],\n",
      "           [-1.9993, -1.9993, -1.9993,  ..., -1.9856, -1.9856, -1.9856],\n",
      "           [-1.9993, -1.9993, -1.9993,  ..., -1.9856, -1.9856, -1.9856]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9988, -1.9988, -1.9987,  ..., -1.9853, -1.9855, -1.9855],\n",
      "           [-1.9987, -1.9987, -1.9987,  ..., -1.9851, -1.9853, -1.9854],\n",
      "           [-1.9987, -1.9987, -1.9987,  ..., -1.9851, -1.9850, -1.9852]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9988, -1.9988, -1.9988,  ..., -1.9835, -1.9834, -1.9834],\n",
      "           [-1.9988, -1.9988, -1.9988,  ..., -1.9835, -1.9835, -1.9835],\n",
      "           [-1.9988, -1.9988, -1.9988,  ..., -1.9834, -1.9834, -1.9834]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9975, -1.9975, -1.9974,  ..., -1.9876, -1.9881, -1.9884],\n",
      "           [-1.9975, -1.9975, -1.9974,  ..., -1.9874, -1.9879, -1.9882],\n",
      "           [-1.9976, -1.9976, -1.9975,  ..., -1.9873, -1.9879, -1.9882]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9952, -1.9951, -1.9951,  ..., -1.9868, -1.9873, -1.9877],\n",
      "           [-1.9940, -1.9939, -1.9938,  ..., -1.9865, -1.9864, -1.9862],\n",
      "           [-1.9929, -1.9927, -1.9925,  ..., -1.9863, -1.9859, -1.9854]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9954, -1.9954, -1.9954,  ..., -1.9890, -1.9875, -1.9885],\n",
      "           [-1.9954, -1.9954, -1.9954,  ..., -1.9892, -1.9890, -1.9906],\n",
      "           [-1.9955, -1.9955, -1.9955,  ..., -1.9902, -1.9908, -1.9925]]],\n",
      "\n",
      "\n",
      "         [[[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9997, -1.9997, -1.9997,  ..., -1.9870, -1.9870, -1.9869],\n",
      "           [-1.9997, -1.9997, -1.9997,  ..., -1.9870, -1.9870, -1.9870],\n",
      "           [-1.9997, -1.9997, -1.9997,  ..., -1.9869, -1.9870, -1.9870]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9993, -1.9993, -1.9992,  ..., -1.9867, -1.9869, -1.9869],\n",
      "           [-1.9992, -1.9992, -1.9992,  ..., -1.9865, -1.9867, -1.9868],\n",
      "           [-1.9991, -1.9992, -1.9992,  ..., -1.9865, -1.9864, -1.9866]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9995, -1.9995, -1.9995,  ..., -1.9847, -1.9847, -1.9847],\n",
      "           [-1.9995, -1.9995, -1.9995,  ..., -1.9847, -1.9847, -1.9847],\n",
      "           [-1.9994, -1.9994, -1.9994,  ..., -1.9847, -1.9846, -1.9846]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9984, -1.9984, -1.9983,  ..., -1.9887, -1.9892, -1.9896],\n",
      "           [-1.9984, -1.9984, -1.9983,  ..., -1.9885, -1.9890, -1.9894],\n",
      "           [-1.9985, -1.9984, -1.9984,  ..., -1.9885, -1.9890, -1.9893]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9954, -1.9954, -1.9953,  ..., -1.9880, -1.9885, -1.9889],\n",
      "           [-1.9937, -1.9936, -1.9935,  ..., -1.9878, -1.9877, -1.9875],\n",
      "           [-1.9923, -1.9921, -1.9919,  ..., -1.9876, -1.9871, -1.9866]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9961, -1.9961, -1.9961,  ..., -1.9905, -1.9890, -1.9900],\n",
      "           [-1.9962, -1.9962, -1.9962,  ..., -1.9907, -1.9905, -1.9921],\n",
      "           [-1.9962, -1.9962, -1.9962,  ..., -1.9916, -1.9922, -1.9939]]]],\n",
      "\n",
      "\n",
      "\n",
      "        [[[[-1.9886, -1.9888, -1.9891,  ..., -1.9918, -1.9914, -1.9912],\n",
      "           [-1.9884, -1.9886, -1.9889,  ..., -1.9923, -1.9921, -1.9919],\n",
      "           [-1.9881, -1.9883, -1.9887,  ..., -1.9931, -1.9929, -1.9928],\n",
      "           ...,\n",
      "           [-1.9979, -1.9983, -1.9988,  ..., -1.9911, -1.9910, -1.9908],\n",
      "           [-1.9979, -1.9983, -1.9988,  ..., -1.9909, -1.9910, -1.9910],\n",
      "           [-1.9978, -1.9982, -1.9987,  ..., -1.9910, -1.9910, -1.9910]],\n",
      "\n",
      "          [[-1.9892, -1.9893, -1.9893,  ..., -1.9903, -1.9903, -1.9903],\n",
      "           [-1.9895, -1.9896, -1.9897,  ..., -1.9905, -1.9905, -1.9905],\n",
      "           [-1.9899, -1.9900, -1.9901,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           ...,\n",
      "           [-1.9984, -1.9987, -1.9990,  ..., -1.9913, -1.9908, -1.9903],\n",
      "           [-1.9985, -1.9987, -1.9991,  ..., -1.9915, -1.9909, -1.9905],\n",
      "           [-1.9983, -1.9986, -1.9989,  ..., -1.9914, -1.9911, -1.9908]],\n",
      "\n",
      "          [[-1.9869, -1.9869, -1.9869,  ..., -1.9900, -1.9900, -1.9900],\n",
      "           [-1.9868, -1.9868, -1.9868,  ..., -1.9903, -1.9903, -1.9903],\n",
      "           [-1.9866, -1.9866, -1.9866,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           ...,\n",
      "           [-1.9887, -1.9887, -1.9887,  ..., -1.9888, -1.9888, -1.9888],\n",
      "           [-1.9887, -1.9887, -1.9887,  ..., -1.9888, -1.9888, -1.9888],\n",
      "           [-1.9899, -1.9899, -1.9899,  ..., -1.9896, -1.9896, -1.9896]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9893, -1.9893, -1.9893,  ..., -1.9917, -1.9917, -1.9917],\n",
      "           [-1.9893, -1.9894, -1.9894,  ..., -1.9921, -1.9921, -1.9921],\n",
      "           [-1.9893, -1.9894, -1.9894,  ..., -1.9926, -1.9926, -1.9926],\n",
      "           ...,\n",
      "           [-1.9882, -1.9884, -1.9887,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9882, -1.9884, -1.9887,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9895, -1.9896, -1.9898,  ..., -1.9894, -1.9894, -1.9894]],\n",
      "\n",
      "          [[-1.9899, -1.9899, -1.9900,  ..., -1.9921, -1.9921, -1.9921],\n",
      "           [-1.9901, -1.9901, -1.9901,  ..., -1.9924, -1.9924, -1.9924],\n",
      "           [-1.9903, -1.9903, -1.9903,  ..., -1.9928, -1.9928, -1.9928],\n",
      "           ...,\n",
      "           [-1.9884, -1.9884, -1.9884,  ..., -1.9914, -1.9912, -1.9910],\n",
      "           [-1.9884, -1.9884, -1.9884,  ..., -1.9914, -1.9911, -1.9910],\n",
      "           [-1.9886, -1.9886, -1.9886,  ..., -1.9923, -1.9921, -1.9920]],\n",
      "\n",
      "          [[-1.9898, -1.9898, -1.9899,  ..., -1.9923, -1.9923, -1.9923],\n",
      "           [-1.9900, -1.9900, -1.9901,  ..., -1.9926, -1.9926, -1.9926],\n",
      "           [-1.9902, -1.9902, -1.9903,  ..., -1.9930, -1.9930, -1.9930],\n",
      "           ...,\n",
      "           [-1.9882, -1.9882, -1.9882,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9882, -1.9882, -1.9882,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9883, -1.9883, -1.9883,  ..., -1.9893, -1.9893, -1.9893]]],\n",
      "\n",
      "\n",
      "         [[[-1.9947, -1.9948, -1.9949,  ..., -1.9936, -1.9933, -1.9930],\n",
      "           [-1.9946, -1.9947, -1.9949,  ..., -1.9940, -1.9938, -1.9936],\n",
      "           [-1.9945, -1.9946, -1.9948,  ..., -1.9945, -1.9943, -1.9942],\n",
      "           ...,\n",
      "           [-1.9982, -1.9986, -1.9991,  ..., -1.9932, -1.9930, -1.9929],\n",
      "           [-1.9982, -1.9986, -1.9991,  ..., -1.9930, -1.9931, -1.9930],\n",
      "           [-1.9981, -1.9985, -1.9990,  ..., -1.9930, -1.9929, -1.9929]],\n",
      "\n",
      "          [[-1.9898, -1.9899, -1.9900,  ..., -1.9902, -1.9902, -1.9902],\n",
      "           [-1.9901, -1.9902, -1.9903,  ..., -1.9904, -1.9904, -1.9904],\n",
      "           [-1.9905, -1.9906, -1.9907,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           ...,\n",
      "           [-1.9986, -1.9987, -1.9987,  ..., -1.9929, -1.9924, -1.9920],\n",
      "           [-1.9987, -1.9987, -1.9988,  ..., -1.9933, -1.9928, -1.9923],\n",
      "           [-1.9987, -1.9987, -1.9988,  ..., -1.9933, -1.9929, -1.9926]],\n",
      "\n",
      "          [[-1.9877, -1.9877, -1.9877,  ..., -1.9895, -1.9895, -1.9895],\n",
      "           [-1.9875, -1.9875, -1.9875,  ..., -1.9899, -1.9899, -1.9899],\n",
      "           [-1.9874, -1.9874, -1.9874,  ..., -1.9903, -1.9903, -1.9903],\n",
      "           ...,\n",
      "           [-1.9906, -1.9906, -1.9906,  ..., -1.9908, -1.9908, -1.9908],\n",
      "           [-1.9907, -1.9907, -1.9907,  ..., -1.9909, -1.9909, -1.9909],\n",
      "           [-1.9919, -1.9919, -1.9919,  ..., -1.9918, -1.9918, -1.9918]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9919, -1.9919, -1.9919,  ..., -1.9939, -1.9939, -1.9939],\n",
      "           [-1.9917, -1.9918, -1.9918,  ..., -1.9942, -1.9942, -1.9942],\n",
      "           [-1.9915, -1.9915, -1.9916,  ..., -1.9944, -1.9944, -1.9944],\n",
      "           ...,\n",
      "           [-1.9900, -1.9902, -1.9905,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           [-1.9900, -1.9902, -1.9905,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           [-1.9913, -1.9915, -1.9916,  ..., -1.9912, -1.9912, -1.9912]],\n",
      "\n",
      "          [[-1.9917, -1.9918, -1.9918,  ..., -1.9943, -1.9943, -1.9943],\n",
      "           [-1.9918, -1.9919, -1.9919,  ..., -1.9946, -1.9946, -1.9946],\n",
      "           [-1.9920, -1.9920, -1.9920,  ..., -1.9950, -1.9950, -1.9950],\n",
      "           ...,\n",
      "           [-1.9901, -1.9901, -1.9901,  ..., -1.9935, -1.9933, -1.9932],\n",
      "           [-1.9901, -1.9901, -1.9901,  ..., -1.9935, -1.9933, -1.9931],\n",
      "           [-1.9903, -1.9903, -1.9903,  ..., -1.9944, -1.9942, -1.9941]],\n",
      "\n",
      "          [[-1.9917, -1.9917, -1.9917,  ..., -1.9937, -1.9937, -1.9937],\n",
      "           [-1.9918, -1.9918, -1.9919,  ..., -1.9940, -1.9940, -1.9940],\n",
      "           [-1.9919, -1.9919, -1.9920,  ..., -1.9942, -1.9942, -1.9942],\n",
      "           ...,\n",
      "           [-1.9900, -1.9900, -1.9900,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           [-1.9900, -1.9900, -1.9900,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           [-1.9900, -1.9900, -1.9900,  ..., -1.9911, -1.9911, -1.9911]]],\n",
      "\n",
      "\n",
      "         [[[-1.9971, -1.9974, -1.9979,  ..., -1.9967, -1.9963, -1.9961],\n",
      "           [-1.9969, -1.9973, -1.9979,  ..., -1.9974, -1.9971, -1.9969],\n",
      "           [-1.9968, -1.9972, -1.9978,  ..., -1.9983, -1.9981, -1.9980],\n",
      "           ...,\n",
      "           [-1.9982, -1.9985, -1.9990,  ..., -1.9971, -1.9967, -1.9964],\n",
      "           [-1.9981, -1.9985, -1.9990,  ..., -1.9969, -1.9967, -1.9965],\n",
      "           [-1.9980, -1.9984, -1.9989,  ..., -1.9965, -1.9964, -1.9963]],\n",
      "\n",
      "          [[-1.9897, -1.9898, -1.9899,  ..., -1.9900, -1.9900, -1.9900],\n",
      "           [-1.9901, -1.9901, -1.9902,  ..., -1.9900, -1.9900, -1.9900],\n",
      "           [-1.9905, -1.9905, -1.9906,  ..., -1.9900, -1.9900, -1.9900],\n",
      "           ...,\n",
      "           [-1.9979, -1.9982, -1.9987,  ..., -1.9980, -1.9974, -1.9970],\n",
      "           [-1.9978, -1.9982, -1.9987,  ..., -1.9977, -1.9973, -1.9969],\n",
      "           [-1.9980, -1.9983, -1.9986,  ..., -1.9972, -1.9969, -1.9967]],\n",
      "\n",
      "          [[-1.9880, -1.9880, -1.9880,  ..., -1.9906, -1.9906, -1.9906],\n",
      "           [-1.9878, -1.9878, -1.9878,  ..., -1.9908, -1.9908, -1.9908],\n",
      "           [-1.9876, -1.9876, -1.9876,  ..., -1.9911, -1.9911, -1.9911],\n",
      "           ...,\n",
      "           [-1.9954, -1.9954, -1.9954,  ..., -1.9948, -1.9948, -1.9948],\n",
      "           [-1.9953, -1.9953, -1.9953,  ..., -1.9950, -1.9950, -1.9950],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9953, -1.9953, -1.9953]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9952, -1.9952, -1.9953,  ..., -1.9965, -1.9965, -1.9965],\n",
      "           [-1.9953, -1.9954, -1.9954,  ..., -1.9970, -1.9970, -1.9970],\n",
      "           [-1.9954, -1.9955, -1.9955,  ..., -1.9975, -1.9975, -1.9975],\n",
      "           ...,\n",
      "           [-1.9951, -1.9953, -1.9955,  ..., -1.9960, -1.9960, -1.9960],\n",
      "           [-1.9951, -1.9953, -1.9955,  ..., -1.9960, -1.9960, -1.9960],\n",
      "           [-1.9957, -1.9958, -1.9960,  ..., -1.9960, -1.9960, -1.9960]],\n",
      "\n",
      "          [[-1.9948, -1.9948, -1.9949,  ..., -1.9986, -1.9986, -1.9986],\n",
      "           [-1.9952, -1.9952, -1.9953,  ..., -1.9983, -1.9983, -1.9983],\n",
      "           [-1.9957, -1.9957, -1.9958,  ..., -1.9979, -1.9979, -1.9979],\n",
      "           ...,\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9966, -1.9964, -1.9963],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9966, -1.9964, -1.9963],\n",
      "           [-1.9955, -1.9955, -1.9955,  ..., -1.9971, -1.9970, -1.9968]],\n",
      "\n",
      "          [[-1.9947, -1.9948, -1.9948,  ..., -1.9956, -1.9956, -1.9956],\n",
      "           [-1.9951, -1.9952, -1.9952,  ..., -1.9961, -1.9961, -1.9961],\n",
      "           [-1.9956, -1.9957, -1.9957,  ..., -1.9968, -1.9968, -1.9968],\n",
      "           ...,\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9958, -1.9958, -1.9958],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9958, -1.9958, -1.9958],\n",
      "           [-1.9954, -1.9954, -1.9954,  ..., -1.9959, -1.9959, -1.9959]]]],\n",
      "\n",
      "\n",
      "\n",
      "        [[[[-1.9886, -1.9888, -1.9891,  ..., -1.9918, -1.9914, -1.9912],\n",
      "           [-1.9884, -1.9886, -1.9889,  ..., -1.9923, -1.9921, -1.9919],\n",
      "           [-1.9881, -1.9883, -1.9887,  ..., -1.9931, -1.9929, -1.9928],\n",
      "           ...,\n",
      "           [-1.9979, -1.9983, -1.9988,  ..., -1.9911, -1.9910, -1.9908],\n",
      "           [-1.9979, -1.9983, -1.9988,  ..., -1.9909, -1.9910, -1.9910],\n",
      "           [-1.9978, -1.9982, -1.9987,  ..., -1.9910, -1.9910, -1.9910]],\n",
      "\n",
      "          [[-1.9892, -1.9893, -1.9893,  ..., -1.9903, -1.9903, -1.9903],\n",
      "           [-1.9895, -1.9896, -1.9897,  ..., -1.9905, -1.9905, -1.9905],\n",
      "           [-1.9899, -1.9900, -1.9901,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           ...,\n",
      "           [-1.9984, -1.9987, -1.9990,  ..., -1.9913, -1.9908, -1.9903],\n",
      "           [-1.9985, -1.9987, -1.9991,  ..., -1.9915, -1.9909, -1.9905],\n",
      "           [-1.9983, -1.9986, -1.9989,  ..., -1.9914, -1.9911, -1.9908]],\n",
      "\n",
      "          [[-1.9869, -1.9869, -1.9869,  ..., -1.9900, -1.9900, -1.9900],\n",
      "           [-1.9868, -1.9868, -1.9868,  ..., -1.9903, -1.9903, -1.9903],\n",
      "           [-1.9866, -1.9866, -1.9866,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           ...,\n",
      "           [-1.9887, -1.9887, -1.9887,  ..., -1.9888, -1.9888, -1.9888],\n",
      "           [-1.9887, -1.9887, -1.9887,  ..., -1.9888, -1.9888, -1.9888],\n",
      "           [-1.9899, -1.9899, -1.9899,  ..., -1.9896, -1.9896, -1.9896]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9893, -1.9893, -1.9893,  ..., -1.9917, -1.9917, -1.9917],\n",
      "           [-1.9893, -1.9894, -1.9894,  ..., -1.9921, -1.9921, -1.9921],\n",
      "           [-1.9893, -1.9894, -1.9894,  ..., -1.9926, -1.9926, -1.9926],\n",
      "           ...,\n",
      "           [-1.9882, -1.9884, -1.9887,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9882, -1.9884, -1.9887,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9895, -1.9896, -1.9898,  ..., -1.9894, -1.9894, -1.9894]],\n",
      "\n",
      "          [[-1.9899, -1.9899, -1.9900,  ..., -1.9921, -1.9921, -1.9921],\n",
      "           [-1.9901, -1.9901, -1.9901,  ..., -1.9924, -1.9924, -1.9924],\n",
      "           [-1.9903, -1.9903, -1.9903,  ..., -1.9928, -1.9928, -1.9928],\n",
      "           ...,\n",
      "           [-1.9884, -1.9884, -1.9884,  ..., -1.9914, -1.9912, -1.9910],\n",
      "           [-1.9884, -1.9884, -1.9884,  ..., -1.9914, -1.9911, -1.9910],\n",
      "           [-1.9886, -1.9886, -1.9886,  ..., -1.9923, -1.9921, -1.9920]],\n",
      "\n",
      "          [[-1.9898, -1.9898, -1.9899,  ..., -1.9923, -1.9923, -1.9923],\n",
      "           [-1.9900, -1.9900, -1.9901,  ..., -1.9926, -1.9926, -1.9926],\n",
      "           [-1.9902, -1.9902, -1.9903,  ..., -1.9930, -1.9930, -1.9930],\n",
      "           ...,\n",
      "           [-1.9882, -1.9882, -1.9882,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9882, -1.9882, -1.9882,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9883, -1.9883, -1.9883,  ..., -1.9893, -1.9893, -1.9893]]],\n",
      "\n",
      "\n",
      "         [[[-1.9947, -1.9948, -1.9949,  ..., -1.9936, -1.9933, -1.9930],\n",
      "           [-1.9946, -1.9947, -1.9949,  ..., -1.9940, -1.9938, -1.9936],\n",
      "           [-1.9945, -1.9946, -1.9948,  ..., -1.9945, -1.9943, -1.9942],\n",
      "           ...,\n",
      "           [-1.9982, -1.9986, -1.9991,  ..., -1.9932, -1.9930, -1.9929],\n",
      "           [-1.9982, -1.9986, -1.9991,  ..., -1.9930, -1.9931, -1.9930],\n",
      "           [-1.9981, -1.9985, -1.9990,  ..., -1.9930, -1.9929, -1.9929]],\n",
      "\n",
      "          [[-1.9898, -1.9899, -1.9900,  ..., -1.9902, -1.9902, -1.9902],\n",
      "           [-1.9901, -1.9902, -1.9903,  ..., -1.9904, -1.9904, -1.9904],\n",
      "           [-1.9905, -1.9906, -1.9907,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           ...,\n",
      "           [-1.9986, -1.9987, -1.9987,  ..., -1.9929, -1.9924, -1.9920],\n",
      "           [-1.9987, -1.9987, -1.9988,  ..., -1.9933, -1.9928, -1.9923],\n",
      "           [-1.9987, -1.9987, -1.9988,  ..., -1.9933, -1.9929, -1.9926]],\n",
      "\n",
      "          [[-1.9877, -1.9877, -1.9877,  ..., -1.9895, -1.9895, -1.9895],\n",
      "           [-1.9875, -1.9875, -1.9875,  ..., -1.9899, -1.9899, -1.9899],\n",
      "           [-1.9874, -1.9874, -1.9874,  ..., -1.9903, -1.9903, -1.9903],\n",
      "           ...,\n",
      "           [-1.9906, -1.9906, -1.9906,  ..., -1.9908, -1.9908, -1.9908],\n",
      "           [-1.9907, -1.9907, -1.9907,  ..., -1.9909, -1.9909, -1.9909],\n",
      "           [-1.9919, -1.9919, -1.9919,  ..., -1.9918, -1.9918, -1.9918]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9919, -1.9919, -1.9919,  ..., -1.9939, -1.9939, -1.9939],\n",
      "           [-1.9917, -1.9918, -1.9918,  ..., -1.9942, -1.9942, -1.9942],\n",
      "           [-1.9915, -1.9915, -1.9916,  ..., -1.9944, -1.9944, -1.9944],\n",
      "           ...,\n",
      "           [-1.9900, -1.9902, -1.9905,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           [-1.9900, -1.9902, -1.9905,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           [-1.9913, -1.9915, -1.9916,  ..., -1.9912, -1.9912, -1.9912]],\n",
      "\n",
      "          [[-1.9917, -1.9918, -1.9918,  ..., -1.9943, -1.9943, -1.9943],\n",
      "           [-1.9918, -1.9919, -1.9919,  ..., -1.9946, -1.9946, -1.9946],\n",
      "           [-1.9920, -1.9920, -1.9920,  ..., -1.9950, -1.9950, -1.9950],\n",
      "           ...,\n",
      "           [-1.9901, -1.9901, -1.9901,  ..., -1.9935, -1.9933, -1.9932],\n",
      "           [-1.9901, -1.9901, -1.9901,  ..., -1.9935, -1.9933, -1.9931],\n",
      "           [-1.9903, -1.9903, -1.9903,  ..., -1.9944, -1.9942, -1.9941]],\n",
      "\n",
      "          [[-1.9917, -1.9917, -1.9917,  ..., -1.9937, -1.9937, -1.9937],\n",
      "           [-1.9918, -1.9918, -1.9919,  ..., -1.9940, -1.9940, -1.9940],\n",
      "           [-1.9919, -1.9919, -1.9920,  ..., -1.9942, -1.9942, -1.9942],\n",
      "           ...,\n",
      "           [-1.9900, -1.9900, -1.9900,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           [-1.9900, -1.9900, -1.9900,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           [-1.9900, -1.9900, -1.9900,  ..., -1.9911, -1.9911, -1.9911]]],\n",
      "\n",
      "\n",
      "         [[[-1.9971, -1.9974, -1.9979,  ..., -1.9967, -1.9963, -1.9961],\n",
      "           [-1.9969, -1.9973, -1.9979,  ..., -1.9974, -1.9971, -1.9969],\n",
      "           [-1.9968, -1.9972, -1.9978,  ..., -1.9983, -1.9981, -1.9980],\n",
      "           ...,\n",
      "           [-1.9982, -1.9985, -1.9990,  ..., -1.9971, -1.9967, -1.9964],\n",
      "           [-1.9981, -1.9985, -1.9990,  ..., -1.9969, -1.9967, -1.9965],\n",
      "           [-1.9980, -1.9984, -1.9989,  ..., -1.9965, -1.9964, -1.9963]],\n",
      "\n",
      "          [[-1.9897, -1.9898, -1.9899,  ..., -1.9900, -1.9900, -1.9900],\n",
      "           [-1.9901, -1.9901, -1.9902,  ..., -1.9900, -1.9900, -1.9900],\n",
      "           [-1.9905, -1.9905, -1.9906,  ..., -1.9900, -1.9900, -1.9900],\n",
      "           ...,\n",
      "           [-1.9979, -1.9982, -1.9987,  ..., -1.9980, -1.9974, -1.9970],\n",
      "           [-1.9978, -1.9982, -1.9987,  ..., -1.9977, -1.9973, -1.9969],\n",
      "           [-1.9980, -1.9983, -1.9986,  ..., -1.9972, -1.9969, -1.9967]],\n",
      "\n",
      "          [[-1.9880, -1.9880, -1.9880,  ..., -1.9906, -1.9906, -1.9906],\n",
      "           [-1.9878, -1.9878, -1.9878,  ..., -1.9908, -1.9908, -1.9908],\n",
      "           [-1.9876, -1.9876, -1.9876,  ..., -1.9911, -1.9911, -1.9911],\n",
      "           ...,\n",
      "           [-1.9954, -1.9954, -1.9954,  ..., -1.9948, -1.9948, -1.9948],\n",
      "           [-1.9953, -1.9953, -1.9953,  ..., -1.9950, -1.9950, -1.9950],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9953, -1.9953, -1.9953]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9952, -1.9952, -1.9953,  ..., -1.9965, -1.9965, -1.9965],\n",
      "           [-1.9953, -1.9954, -1.9954,  ..., -1.9970, -1.9970, -1.9970],\n",
      "           [-1.9954, -1.9955, -1.9955,  ..., -1.9975, -1.9975, -1.9975],\n",
      "           ...,\n",
      "           [-1.9951, -1.9953, -1.9955,  ..., -1.9960, -1.9960, -1.9960],\n",
      "           [-1.9951, -1.9953, -1.9955,  ..., -1.9960, -1.9960, -1.9960],\n",
      "           [-1.9957, -1.9958, -1.9960,  ..., -1.9960, -1.9960, -1.9960]],\n",
      "\n",
      "          [[-1.9948, -1.9948, -1.9949,  ..., -1.9986, -1.9986, -1.9986],\n",
      "           [-1.9952, -1.9952, -1.9953,  ..., -1.9983, -1.9983, -1.9983],\n",
      "           [-1.9957, -1.9957, -1.9958,  ..., -1.9979, -1.9979, -1.9979],\n",
      "           ...,\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9966, -1.9964, -1.9963],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9966, -1.9964, -1.9963],\n",
      "           [-1.9955, -1.9955, -1.9955,  ..., -1.9971, -1.9970, -1.9968]],\n",
      "\n",
      "          [[-1.9947, -1.9948, -1.9948,  ..., -1.9956, -1.9956, -1.9956],\n",
      "           [-1.9951, -1.9952, -1.9952,  ..., -1.9961, -1.9961, -1.9961],\n",
      "           [-1.9956, -1.9957, -1.9957,  ..., -1.9968, -1.9968, -1.9968],\n",
      "           ...,\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9958, -1.9958, -1.9958],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9958, -1.9958, -1.9958],\n",
      "           [-1.9954, -1.9954, -1.9954,  ..., -1.9959, -1.9959, -1.9959]]]]]), tensor([[[[[-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
      "           [-1.9999, -1.9999, -1.9990,  ..., -1.9994, -1.9998, -1.9998],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
      "           [-1.9999, -1.9999, -1.9990,  ..., -1.9994, -1.9998, -1.9998],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9999, -1.9999, -1.9999],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9999, -1.9999, -1.9999],\n",
      "           [-2.0000, -2.0000, -1.9991,  ..., -1.9995, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9973, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9973, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9973, -1.9997, -1.9997]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9998, -1.9998, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-1.9998, -1.9998, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-1.9998, -1.9998, -1.9998,  ..., -1.9991, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9990, -1.9997, -1.9997],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9997, -1.9997]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9990, -1.9997, -1.9997],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9997, -1.9997]]],\n",
      "\n",
      "\n",
      "         [[[-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
      "           [-1.9998, -1.9998, -1.9990,  ..., -1.9986, -1.9997, -1.9997],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9965, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9967, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
      "           [-1.9998, -1.9998, -1.9990,  ..., -1.9986, -1.9997, -1.9997],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9965, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9967, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9999,  ..., -1.9999, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9999,  ..., -1.9999, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9990,  ..., -1.9990, -2.0000, -2.0000],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9965, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9966, -1.9998, -1.9998]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9995, -1.9995, -1.9996,  ..., -1.9994, -1.9999, -1.9999],\n",
      "           [-1.9995, -1.9995, -1.9996,  ..., -1.9994, -1.9999, -1.9999],\n",
      "           [-1.9996, -1.9996, -1.9996,  ..., -1.9988, -1.9997, -1.9997],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9987, -1.9996, -1.9996],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9996, -1.9996]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9987, -1.9996, -1.9996],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9996, -1.9996]]],\n",
      "\n",
      "\n",
      "         [[[-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
      "           [-1.9998, -1.9998, -1.9989,  ..., -1.9987, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9964, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9965, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -1.9998, -1.9998]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
      "           [-1.9998, -1.9998, -1.9989,  ..., -1.9987, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9964, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9965, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -1.9998, -1.9998]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9990,  ..., -1.9989, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-1.9998, -1.9998, -1.9996,  ..., -1.9965, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9966, -1.9998, -1.9998]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9993, -1.9993, -1.9994,  ..., -1.9989, -1.9995, -1.9995],\n",
      "           [-1.9993, -1.9993, -1.9994,  ..., -1.9989, -1.9995, -1.9995],\n",
      "           [-1.9994, -1.9994, -1.9994,  ..., -1.9983, -1.9993, -1.9993],\n",
      "           ...,\n",
      "           [-1.9997, -1.9997, -1.9997,  ..., -1.9959, -1.9997, -1.9997],\n",
      "           [-1.9997, -1.9997, -1.9997,  ..., -1.9959, -1.9997, -1.9997],\n",
      "           [-1.9997, -1.9997, -1.9997,  ..., -1.9959, -1.9997, -1.9997]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
      "           [-1.9998, -1.9998, -1.9995,  ..., -1.9983, -1.9992, -1.9992],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9995, -1.9995],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
      "           [-1.9998, -1.9998, -1.9995,  ..., -1.9983, -1.9992, -1.9992],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9995, -1.9995],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994]]]],\n",
      "\n",
      "\n",
      "\n",
      "        [[[[-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
      "           [-1.9999, -1.9999, -1.9990,  ..., -1.9994, -1.9998, -1.9998],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9998, -1.9998, -1.9998],\n",
      "           [-1.9999, -1.9999, -1.9990,  ..., -1.9994, -1.9998, -1.9998],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9975, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9999, -1.9999, -1.9999],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9999, -1.9999, -1.9999],\n",
      "           [-2.0000, -2.0000, -1.9991,  ..., -1.9995, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9973, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9973, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9973, -1.9997, -1.9997]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9998, -1.9998, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-1.9998, -1.9998, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-1.9998, -1.9998, -1.9998,  ..., -1.9991, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9990, -1.9997, -1.9997],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9997, -1.9997]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9997, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9990, -1.9997, -1.9997],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9961, -1.9997, -1.9997]]],\n",
      "\n",
      "\n",
      "         [[[-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
      "           [-1.9998, -1.9998, -1.9990,  ..., -1.9986, -1.9997, -1.9997],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9965, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9967, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9999,  ..., -1.9996, -1.9999, -1.9999],\n",
      "           [-1.9998, -1.9998, -1.9990,  ..., -1.9986, -1.9997, -1.9997],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9965, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9967, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9999,  ..., -1.9999, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9999,  ..., -1.9999, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9990,  ..., -1.9990, -2.0000, -2.0000],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9965, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9966, -1.9998, -1.9998]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9995, -1.9995, -1.9996,  ..., -1.9994, -1.9999, -1.9999],\n",
      "           [-1.9995, -1.9995, -1.9996,  ..., -1.9994, -1.9999, -1.9999],\n",
      "           [-1.9996, -1.9996, -1.9996,  ..., -1.9988, -1.9997, -1.9997],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9999,  ..., -1.9963, -2.0000, -2.0000]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9987, -1.9996, -1.9996],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9996, -1.9996]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9994, -1.9999, -1.9999],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9987, -1.9996, -1.9996],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9998, -1.9998],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9998,  ..., -1.9960, -1.9996, -1.9996]]],\n",
      "\n",
      "\n",
      "         [[[-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
      "           [-1.9998, -1.9998, -1.9989,  ..., -1.9987, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9964, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9965, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -1.9998, -1.9998]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9996, -2.0000, -2.0000],\n",
      "           [-1.9998, -1.9998, -1.9989,  ..., -1.9987, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9964, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9965, -1.9997, -1.9997],\n",
      "           [-2.0000, -2.0000, -1.9997,  ..., -1.9966, -1.9998, -1.9998]],\n",
      "\n",
      "          [[-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -2.0000,  ..., -1.9999, -2.0000, -2.0000],\n",
      "           [-2.0000, -2.0000, -1.9990,  ..., -1.9989, -1.9999, -1.9999],\n",
      "           ...,\n",
      "           [-1.9998, -1.9998, -1.9996,  ..., -1.9965, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9966, -2.0000, -2.0000],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9966, -1.9998, -1.9998]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9993, -1.9993, -1.9994,  ..., -1.9989, -1.9995, -1.9995],\n",
      "           [-1.9993, -1.9993, -1.9994,  ..., -1.9989, -1.9995, -1.9995],\n",
      "           [-1.9994, -1.9994, -1.9994,  ..., -1.9983, -1.9993, -1.9993],\n",
      "           ...,\n",
      "           [-1.9997, -1.9997, -1.9997,  ..., -1.9959, -1.9997, -1.9997],\n",
      "           [-1.9997, -1.9997, -1.9997,  ..., -1.9959, -1.9997, -1.9997],\n",
      "           [-1.9997, -1.9997, -1.9997,  ..., -1.9959, -1.9997, -1.9997]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
      "           [-1.9998, -1.9998, -1.9995,  ..., -1.9983, -1.9992, -1.9992],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9995, -1.9995],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994]],\n",
      "\n",
      "          [[-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
      "           [-1.9999, -1.9999, -1.9996,  ..., -1.9989, -1.9995, -1.9995],\n",
      "           [-1.9998, -1.9998, -1.9995,  ..., -1.9983, -1.9992, -1.9992],\n",
      "           ...,\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9995, -1.9995],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994],\n",
      "           [-1.9999, -1.9999, -1.9997,  ..., -1.9956, -1.9994, -1.9994]]]],\n",
      "\n",
      "\n",
      "\n",
      "        [[[[-1.9914, -1.9918, -1.9923,  ..., -1.9882, -1.9883, -1.9884],\n",
      "           [-1.9917, -1.9920, -1.9924,  ..., -1.9884, -1.9885, -1.9886],\n",
      "           [-1.9919, -1.9921, -1.9924,  ..., -1.9886, -1.9888, -1.9889],\n",
      "           ...,\n",
      "           [-1.9930, -1.9930, -1.9930,  ..., -1.9926, -1.9927, -1.9926],\n",
      "           [-1.9932, -1.9933, -1.9934,  ..., -1.9925, -1.9927, -1.9926],\n",
      "           [-1.9935, -1.9936, -1.9937,  ..., -1.9926, -1.9928, -1.9926]],\n",
      "\n",
      "          [[-1.9914, -1.9918, -1.9923,  ..., -1.9882, -1.9883, -1.9884],\n",
      "           [-1.9917, -1.9920, -1.9924,  ..., -1.9884, -1.9885, -1.9886],\n",
      "           [-1.9919, -1.9921, -1.9924,  ..., -1.9886, -1.9888, -1.9889],\n",
      "           ...,\n",
      "           [-1.9930, -1.9930, -1.9930,  ..., -1.9926, -1.9927, -1.9926],\n",
      "           [-1.9932, -1.9933, -1.9934,  ..., -1.9925, -1.9927, -1.9926],\n",
      "           [-1.9935, -1.9936, -1.9937,  ..., -1.9926, -1.9928, -1.9926]],\n",
      "\n",
      "          [[-1.9917, -1.9919, -1.9923,  ..., -1.9884, -1.9885, -1.9886],\n",
      "           [-1.9919, -1.9920, -1.9924,  ..., -1.9885, -1.9886, -1.9888],\n",
      "           [-1.9919, -1.9920, -1.9923,  ..., -1.9886, -1.9889, -1.9891],\n",
      "           ...,\n",
      "           [-1.9937, -1.9936, -1.9935,  ..., -1.9847, -1.9847, -1.9846],\n",
      "           [-1.9940, -1.9939, -1.9939,  ..., -1.9845, -1.9845, -1.9845],\n",
      "           [-1.9939, -1.9940, -1.9941,  ..., -1.9847, -1.9845, -1.9845]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9919, -1.9920, -1.9921,  ..., -1.9886, -1.9886, -1.9886],\n",
      "           [-1.9919, -1.9920, -1.9921,  ..., -1.9885, -1.9885, -1.9885],\n",
      "           [-1.9919, -1.9920, -1.9921,  ..., -1.9885, -1.9885, -1.9885],\n",
      "           ...,\n",
      "           [-1.9942, -1.9937, -1.9931,  ..., -1.9914, -1.9914, -1.9914],\n",
      "           [-1.9942, -1.9937, -1.9931,  ..., -1.9904, -1.9904, -1.9904],\n",
      "           [-1.9942, -1.9937, -1.9931,  ..., -1.9896, -1.9896, -1.9896]],\n",
      "\n",
      "          [[-1.9918, -1.9919, -1.9920,  ..., -1.9886, -1.9886, -1.9886],\n",
      "           [-1.9918, -1.9919, -1.9920,  ..., -1.9885, -1.9885, -1.9885],\n",
      "           [-1.9918, -1.9919, -1.9920,  ..., -1.9885, -1.9885, -1.9885],\n",
      "           ...,\n",
      "           [-1.9947, -1.9937, -1.9925,  ..., -1.9914, -1.9914, -1.9914],\n",
      "           [-1.9947, -1.9937, -1.9925,  ..., -1.9904, -1.9904, -1.9904],\n",
      "           [-1.9947, -1.9937, -1.9925,  ..., -1.9896, -1.9896, -1.9896]],\n",
      "\n",
      "          [[-1.9918, -1.9919, -1.9920,  ..., -1.9886, -1.9886, -1.9886],\n",
      "           [-1.9918, -1.9919, -1.9920,  ..., -1.9885, -1.9885, -1.9885],\n",
      "           [-1.9918, -1.9919, -1.9920,  ..., -1.9885, -1.9885, -1.9885],\n",
      "           ...,\n",
      "           [-1.9947, -1.9937, -1.9925,  ..., -1.9914, -1.9914, -1.9914],\n",
      "           [-1.9947, -1.9937, -1.9925,  ..., -1.9904, -1.9904, -1.9904],\n",
      "           [-1.9947, -1.9937, -1.9925,  ..., -1.9896, -1.9896, -1.9896]]],\n",
      "\n",
      "\n",
      "         [[[-1.9919, -1.9924, -1.9929,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9921, -1.9925, -1.9930,  ..., -1.9894, -1.9895, -1.9895],\n",
      "           [-1.9923, -1.9926, -1.9930,  ..., -1.9898, -1.9899, -1.9899],\n",
      "           ...,\n",
      "           [-1.9945, -1.9945, -1.9946,  ..., -1.9937, -1.9939, -1.9938],\n",
      "           [-1.9949, -1.9949, -1.9950,  ..., -1.9934, -1.9936, -1.9935],\n",
      "           [-1.9950, -1.9951, -1.9952,  ..., -1.9931, -1.9933, -1.9931]],\n",
      "\n",
      "          [[-1.9919, -1.9924, -1.9929,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9921, -1.9925, -1.9930,  ..., -1.9894, -1.9895, -1.9895],\n",
      "           [-1.9923, -1.9926, -1.9930,  ..., -1.9898, -1.9899, -1.9899],\n",
      "           ...,\n",
      "           [-1.9945, -1.9945, -1.9946,  ..., -1.9937, -1.9939, -1.9938],\n",
      "           [-1.9949, -1.9949, -1.9950,  ..., -1.9934, -1.9936, -1.9935],\n",
      "           [-1.9950, -1.9951, -1.9952,  ..., -1.9931, -1.9933, -1.9931]],\n",
      "\n",
      "          [[-1.9922, -1.9925, -1.9929,  ..., -1.9888, -1.9887, -1.9888],\n",
      "           [-1.9924, -1.9926, -1.9930,  ..., -1.9893, -1.9894, -1.9894],\n",
      "           [-1.9925, -1.9925, -1.9929,  ..., -1.9896, -1.9898, -1.9898],\n",
      "           ...,\n",
      "           [-1.9945, -1.9945, -1.9946,  ..., -1.9857, -1.9859, -1.9860],\n",
      "           [-1.9948, -1.9949, -1.9951,  ..., -1.9854, -1.9856, -1.9857],\n",
      "           [-1.9951, -1.9951, -1.9953,  ..., -1.9854, -1.9855, -1.9854]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9928, -1.9929, -1.9930,  ..., -1.9890, -1.9890, -1.9890],\n",
      "           [-1.9928, -1.9929, -1.9930,  ..., -1.9891, -1.9891, -1.9891],\n",
      "           [-1.9928, -1.9929, -1.9930,  ..., -1.9893, -1.9893, -1.9893],\n",
      "           ...,\n",
      "           [-1.9952, -1.9947, -1.9941,  ..., -1.9925, -1.9925, -1.9925],\n",
      "           [-1.9952, -1.9947, -1.9941,  ..., -1.9915, -1.9915, -1.9915],\n",
      "           [-1.9952, -1.9947, -1.9941,  ..., -1.9907, -1.9907, -1.9907]],\n",
      "\n",
      "          [[-1.9928, -1.9928, -1.9929,  ..., -1.9890, -1.9890, -1.9890],\n",
      "           [-1.9928, -1.9928, -1.9929,  ..., -1.9891, -1.9891, -1.9891],\n",
      "           [-1.9928, -1.9928, -1.9929,  ..., -1.9893, -1.9893, -1.9893],\n",
      "           ...,\n",
      "           [-1.9953, -1.9943, -1.9931,  ..., -1.9925, -1.9925, -1.9925],\n",
      "           [-1.9953, -1.9943, -1.9931,  ..., -1.9915, -1.9915, -1.9915],\n",
      "           [-1.9953, -1.9943, -1.9931,  ..., -1.9907, -1.9907, -1.9907]],\n",
      "\n",
      "          [[-1.9928, -1.9928, -1.9929,  ..., -1.9890, -1.9890, -1.9890],\n",
      "           [-1.9928, -1.9928, -1.9929,  ..., -1.9891, -1.9891, -1.9891],\n",
      "           [-1.9928, -1.9928, -1.9929,  ..., -1.9893, -1.9893, -1.9893],\n",
      "           ...,\n",
      "           [-1.9953, -1.9943, -1.9931,  ..., -1.9925, -1.9925, -1.9925],\n",
      "           [-1.9953, -1.9943, -1.9931,  ..., -1.9915, -1.9915, -1.9915],\n",
      "           [-1.9953, -1.9943, -1.9931,  ..., -1.9907, -1.9907, -1.9907]]],\n",
      "\n",
      "\n",
      "         [[[-1.9930, -1.9934, -1.9940,  ..., -1.9897, -1.9898, -1.9900],\n",
      "           [-1.9932, -1.9936, -1.9940,  ..., -1.9903, -1.9905, -1.9905],\n",
      "           [-1.9934, -1.9937, -1.9940,  ..., -1.9909, -1.9910, -1.9911],\n",
      "           ...,\n",
      "           [-1.9954, -1.9954, -1.9954,  ..., -1.9948, -1.9949, -1.9948],\n",
      "           [-1.9957, -1.9958, -1.9959,  ..., -1.9946, -1.9948, -1.9947],\n",
      "           [-1.9960, -1.9961, -1.9962,  ..., -1.9944, -1.9946, -1.9945]],\n",
      "\n",
      "          [[-1.9930, -1.9934, -1.9940,  ..., -1.9897, -1.9898, -1.9900],\n",
      "           [-1.9932, -1.9936, -1.9940,  ..., -1.9903, -1.9905, -1.9905],\n",
      "           [-1.9934, -1.9937, -1.9940,  ..., -1.9909, -1.9910, -1.9911],\n",
      "           ...,\n",
      "           [-1.9954, -1.9954, -1.9954,  ..., -1.9948, -1.9949, -1.9948],\n",
      "           [-1.9957, -1.9958, -1.9959,  ..., -1.9946, -1.9948, -1.9947],\n",
      "           [-1.9960, -1.9961, -1.9962,  ..., -1.9944, -1.9946, -1.9945]],\n",
      "\n",
      "          [[-1.9931, -1.9934, -1.9938,  ..., -1.9900, -1.9900, -1.9901],\n",
      "           [-1.9933, -1.9935, -1.9938,  ..., -1.9905, -1.9905, -1.9906],\n",
      "           [-1.9934, -1.9934, -1.9938,  ..., -1.9909, -1.9910, -1.9912],\n",
      "           ...,\n",
      "           [-1.9957, -1.9956, -1.9956,  ..., -1.9871, -1.9873, -1.9874],\n",
      "           [-1.9960, -1.9960, -1.9961,  ..., -1.9868, -1.9869, -1.9870],\n",
      "           [-1.9961, -1.9962, -1.9963,  ..., -1.9866, -1.9866, -1.9866]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9937, -1.9938, -1.9939,  ..., -1.9908, -1.9908, -1.9908],\n",
      "           [-1.9937, -1.9938, -1.9939,  ..., -1.9909, -1.9909, -1.9909],\n",
      "           [-1.9937, -1.9938, -1.9939,  ..., -1.9910, -1.9910, -1.9910],\n",
      "           ...,\n",
      "           [-1.9954, -1.9949, -1.9943,  ..., -1.9935, -1.9935, -1.9935],\n",
      "           [-1.9954, -1.9949, -1.9943,  ..., -1.9925, -1.9925, -1.9925],\n",
      "           [-1.9954, -1.9949, -1.9943,  ..., -1.9917, -1.9917, -1.9917]],\n",
      "\n",
      "          [[-1.9936, -1.9937, -1.9938,  ..., -1.9908, -1.9908, -1.9908],\n",
      "           [-1.9936, -1.9937, -1.9938,  ..., -1.9909, -1.9909, -1.9909],\n",
      "           [-1.9936, -1.9937, -1.9938,  ..., -1.9910, -1.9910, -1.9910],\n",
      "           ...,\n",
      "           [-1.9956, -1.9947, -1.9934,  ..., -1.9935, -1.9935, -1.9935],\n",
      "           [-1.9956, -1.9947, -1.9934,  ..., -1.9925, -1.9925, -1.9925],\n",
      "           [-1.9956, -1.9947, -1.9934,  ..., -1.9917, -1.9917, -1.9917]],\n",
      "\n",
      "          [[-1.9936, -1.9937, -1.9938,  ..., -1.9908, -1.9908, -1.9908],\n",
      "           [-1.9936, -1.9937, -1.9938,  ..., -1.9909, -1.9909, -1.9909],\n",
      "           [-1.9936, -1.9937, -1.9938,  ..., -1.9910, -1.9910, -1.9910],\n",
      "           ...,\n",
      "           [-1.9956, -1.9947, -1.9934,  ..., -1.9935, -1.9935, -1.9935],\n",
      "           [-1.9956, -1.9947, -1.9934,  ..., -1.9925, -1.9925, -1.9925],\n",
      "           [-1.9956, -1.9947, -1.9934,  ..., -1.9917, -1.9917, -1.9917]]]],\n",
      "\n",
      "\n",
      "\n",
      "        ...,\n",
      "\n",
      "\n",
      "\n",
      "        [[[[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9995, -1.9995, -1.9995,  ..., -1.9845, -1.9845, -1.9844],\n",
      "           [-1.9995, -1.9995, -1.9995,  ..., -1.9845, -1.9845, -1.9844],\n",
      "           [-1.9995, -1.9995, -1.9995,  ..., -1.9844, -1.9844, -1.9845]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9994, -1.9994, -1.9994,  ..., -1.9858, -1.9856, -1.9854],\n",
      "           [-1.9994, -1.9994, -1.9994,  ..., -1.9855, -1.9855, -1.9852],\n",
      "           [-1.9994, -1.9994, -1.9994,  ..., -1.9852, -1.9854, -1.9853]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9994, -1.9994, -1.9993,  ..., -1.9848, -1.9846, -1.9844],\n",
      "           [-1.9995, -1.9995, -1.9994,  ..., -1.9846, -1.9844, -1.9843],\n",
      "           [-1.9993, -1.9994, -1.9995,  ..., -1.9843, -1.9842, -1.9843]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9961, -1.9961, -1.9961,  ..., -1.9961, -1.9956, -1.9953],\n",
      "           [-1.9962, -1.9962, -1.9962,  ..., -1.9965, -1.9961, -1.9957],\n",
      "           [-1.9963, -1.9963, -1.9963,  ..., -1.9967, -1.9966, -1.9962]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9878, -1.9863, -1.9873],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9879, -1.9877, -1.9893],\n",
      "           [-1.9957, -1.9957, -1.9957,  ..., -1.9889, -1.9895, -1.9912]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9955, -1.9955, -1.9955,  ..., -1.9942, -1.9945, -1.9947],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9945, -1.9949, -1.9951],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9945, -1.9950, -1.9953]]],\n",
      "\n",
      "\n",
      "         [[[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9993, -1.9993, -1.9993,  ..., -1.9857, -1.9856, -1.9856],\n",
      "           [-1.9993, -1.9993, -1.9993,  ..., -1.9856, -1.9856, -1.9856],\n",
      "           [-1.9993, -1.9993, -1.9993,  ..., -1.9856, -1.9856, -1.9856]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9992, -1.9992, -1.9992,  ..., -1.9868, -1.9867, -1.9864],\n",
      "           [-1.9992, -1.9992, -1.9992,  ..., -1.9866, -1.9865, -1.9862],\n",
      "           [-1.9992, -1.9992, -1.9992,  ..., -1.9863, -1.9864, -1.9863]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9991, -1.9991, -1.9990,  ..., -1.9860, -1.9858, -1.9856],\n",
      "           [-1.9991, -1.9991, -1.9991,  ..., -1.9858, -1.9856, -1.9855],\n",
      "           [-1.9989, -1.9990, -1.9992,  ..., -1.9856, -1.9854, -1.9855]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9964, -1.9960, -1.9956],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9968, -1.9965, -1.9960],\n",
      "           [-1.9957, -1.9957, -1.9957,  ..., -1.9970, -1.9969, -1.9965]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9954, -1.9954, -1.9954,  ..., -1.9890, -1.9875, -1.9885],\n",
      "           [-1.9954, -1.9954, -1.9954,  ..., -1.9892, -1.9890, -1.9906],\n",
      "           [-1.9955, -1.9955, -1.9955,  ..., -1.9902, -1.9908, -1.9925]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9953, -1.9953, -1.9953,  ..., -1.9950, -1.9953, -1.9955],\n",
      "           [-1.9954, -1.9954, -1.9954,  ..., -1.9953, -1.9958, -1.9960],\n",
      "           [-1.9955, -1.9955, -1.9955,  ..., -1.9954, -1.9959, -1.9961]]],\n",
      "\n",
      "\n",
      "         [[[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9997, -1.9997, -1.9997,  ..., -1.9870, -1.9870, -1.9869],\n",
      "           [-1.9997, -1.9997, -1.9997,  ..., -1.9870, -1.9870, -1.9870],\n",
      "           [-1.9997, -1.9997, -1.9997,  ..., -1.9869, -1.9870, -1.9870]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9996, -1.9996, -1.9996,  ..., -1.9882, -1.9881, -1.9879],\n",
      "           [-1.9996, -1.9996, -1.9996,  ..., -1.9880, -1.9879, -1.9876],\n",
      "           [-1.9996, -1.9996, -1.9996,  ..., -1.9876, -1.9877, -1.9876]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9995, -1.9996, -1.9995,  ..., -1.9873, -1.9872, -1.9870],\n",
      "           [-1.9996, -1.9996, -1.9996,  ..., -1.9872, -1.9870, -1.9869],\n",
      "           [-1.9994, -1.9995, -1.9997,  ..., -1.9869, -1.9868, -1.9869]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9965, -1.9965, -1.9965,  ..., -1.9977, -1.9973, -1.9969],\n",
      "           [-1.9965, -1.9965, -1.9965,  ..., -1.9981, -1.9978, -1.9973],\n",
      "           [-1.9966, -1.9966, -1.9966,  ..., -1.9982, -1.9981, -1.9977]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9961, -1.9961, -1.9961,  ..., -1.9905, -1.9890, -1.9900],\n",
      "           [-1.9962, -1.9962, -1.9962,  ..., -1.9907, -1.9905, -1.9921],\n",
      "           [-1.9962, -1.9962, -1.9962,  ..., -1.9916, -1.9922, -1.9939]],\n",
      "\n",
      "          [[-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           [-1.9826, -1.9826, -1.9826,  ..., -1.9826, -1.9826, -1.9826],\n",
      "           ...,\n",
      "           [-1.9961, -1.9961, -1.9961,  ..., -1.9966, -1.9969, -1.9971],\n",
      "           [-1.9961, -1.9961, -1.9961,  ..., -1.9970, -1.9975, -1.9977],\n",
      "           [-1.9962, -1.9962, -1.9962,  ..., -1.9971, -1.9976, -1.9978]]]],\n",
      "\n",
      "\n",
      "\n",
      "        [[[[-1.9886, -1.9888, -1.9891,  ..., -1.9918, -1.9914, -1.9912],\n",
      "           [-1.9884, -1.9886, -1.9889,  ..., -1.9923, -1.9921, -1.9919],\n",
      "           [-1.9881, -1.9883, -1.9887,  ..., -1.9931, -1.9929, -1.9928],\n",
      "           ...,\n",
      "           [-1.9979, -1.9983, -1.9988,  ..., -1.9911, -1.9910, -1.9908],\n",
      "           [-1.9979, -1.9983, -1.9988,  ..., -1.9909, -1.9910, -1.9910],\n",
      "           [-1.9978, -1.9982, -1.9987,  ..., -1.9910, -1.9910, -1.9910]],\n",
      "\n",
      "          [[-1.9911, -1.9914, -1.9917,  ..., -1.9925, -1.9921, -1.9918],\n",
      "           [-1.9915, -1.9917, -1.9920,  ..., -1.9927, -1.9923, -1.9920],\n",
      "           [-1.9919, -1.9921, -1.9924,  ..., -1.9929, -1.9925, -1.9922],\n",
      "           ...,\n",
      "           [-1.9984, -1.9985, -1.9988,  ..., -1.9908, -1.9910, -1.9910],\n",
      "           [-1.9985, -1.9987, -1.9989,  ..., -1.9912, -1.9909, -1.9907],\n",
      "           [-1.9983, -1.9985, -1.9988,  ..., -1.9910, -1.9907, -1.9906]],\n",
      "\n",
      "          [[-1.9908, -1.9910, -1.9912,  ..., -1.9919, -1.9914, -1.9909],\n",
      "           [-1.9907, -1.9910, -1.9912,  ..., -1.9923, -1.9915, -1.9909],\n",
      "           [-1.9907, -1.9909, -1.9913,  ..., -1.9927, -1.9918, -1.9910],\n",
      "           ...,\n",
      "           [-1.9985, -1.9988, -1.9992,  ..., -1.9909, -1.9910, -1.9911],\n",
      "           [-1.9982, -1.9987, -1.9993,  ..., -1.9908, -1.9910, -1.9911],\n",
      "           [-1.9979, -1.9985, -1.9992,  ..., -1.9908, -1.9909, -1.9911]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9897, -1.9898, -1.9898,  ..., -1.9932, -1.9932, -1.9932],\n",
      "           [-1.9898, -1.9898, -1.9899,  ..., -1.9936, -1.9936, -1.9936],\n",
      "           [-1.9899, -1.9899, -1.9900,  ..., -1.9941, -1.9941, -1.9941],\n",
      "           ...,\n",
      "           [-1.9883, -1.9883, -1.9883,  ..., -1.9887, -1.9886, -1.9885],\n",
      "           [-1.9883, -1.9883, -1.9883,  ..., -1.9887, -1.9886, -1.9885],\n",
      "           [-1.9882, -1.9882, -1.9882,  ..., -1.9890, -1.9889, -1.9888]],\n",
      "\n",
      "          [[-1.9898, -1.9898, -1.9899,  ..., -1.9923, -1.9923, -1.9923],\n",
      "           [-1.9900, -1.9900, -1.9901,  ..., -1.9926, -1.9926, -1.9926],\n",
      "           [-1.9902, -1.9902, -1.9903,  ..., -1.9930, -1.9930, -1.9930],\n",
      "           ...,\n",
      "           [-1.9882, -1.9882, -1.9882,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9882, -1.9882, -1.9882,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9883, -1.9883, -1.9883,  ..., -1.9893, -1.9893, -1.9893]],\n",
      "\n",
      "          [[-1.9892, -1.9892, -1.9893,  ..., -1.9915, -1.9915, -1.9915],\n",
      "           [-1.9896, -1.9896, -1.9896,  ..., -1.9919, -1.9919, -1.9919],\n",
      "           [-1.9900, -1.9900, -1.9901,  ..., -1.9924, -1.9924, -1.9924],\n",
      "           ...,\n",
      "           [-1.9887, -1.9887, -1.9887,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9887, -1.9887, -1.9887,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9886, -1.9886, -1.9886,  ..., -1.9893, -1.9893, -1.9893]]],\n",
      "\n",
      "\n",
      "         [[[-1.9947, -1.9948, -1.9949,  ..., -1.9936, -1.9933, -1.9930],\n",
      "           [-1.9946, -1.9947, -1.9949,  ..., -1.9940, -1.9938, -1.9936],\n",
      "           [-1.9945, -1.9946, -1.9948,  ..., -1.9945, -1.9943, -1.9942],\n",
      "           ...,\n",
      "           [-1.9982, -1.9986, -1.9991,  ..., -1.9932, -1.9930, -1.9929],\n",
      "           [-1.9982, -1.9986, -1.9991,  ..., -1.9930, -1.9931, -1.9930],\n",
      "           [-1.9981, -1.9985, -1.9990,  ..., -1.9930, -1.9929, -1.9929]],\n",
      "\n",
      "          [[-1.9928, -1.9931, -1.9935,  ..., -1.9940, -1.9936, -1.9933],\n",
      "           [-1.9929, -1.9932, -1.9936,  ..., -1.9943, -1.9939, -1.9936],\n",
      "           [-1.9931, -1.9933, -1.9937,  ..., -1.9948, -1.9943, -1.9940],\n",
      "           ...,\n",
      "           [-1.9986, -1.9988, -1.9991,  ..., -1.9928, -1.9927, -1.9926],\n",
      "           [-1.9988, -1.9989, -1.9991,  ..., -1.9931, -1.9926, -1.9922],\n",
      "           [-1.9987, -1.9988, -1.9991,  ..., -1.9927, -1.9923, -1.9921]],\n",
      "\n",
      "          [[-1.9917, -1.9920, -1.9922,  ..., -1.9930, -1.9925, -1.9921],\n",
      "           [-1.9918, -1.9921, -1.9924,  ..., -1.9931, -1.9925, -1.9920],\n",
      "           [-1.9920, -1.9922, -1.9925,  ..., -1.9931, -1.9925, -1.9919],\n",
      "           ...,\n",
      "           [-1.9990, -1.9992, -1.9994,  ..., -1.9930, -1.9927, -1.9926],\n",
      "           [-1.9987, -1.9991, -1.9995,  ..., -1.9930, -1.9929, -1.9928],\n",
      "           [-1.9987, -1.9990, -1.9994,  ..., -1.9929, -1.9929, -1.9929]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9916, -1.9916, -1.9917,  ..., -1.9943, -1.9943, -1.9943],\n",
      "           [-1.9916, -1.9916, -1.9917,  ..., -1.9948, -1.9948, -1.9948],\n",
      "           [-1.9916, -1.9916, -1.9917,  ..., -1.9954, -1.9954, -1.9954],\n",
      "           ...,\n",
      "           [-1.9900, -1.9900, -1.9900,  ..., -1.9905, -1.9904, -1.9903],\n",
      "           [-1.9900, -1.9900, -1.9900,  ..., -1.9905, -1.9904, -1.9903],\n",
      "           [-1.9899, -1.9899, -1.9899,  ..., -1.9909, -1.9907, -1.9906]],\n",
      "\n",
      "          [[-1.9917, -1.9917, -1.9917,  ..., -1.9937, -1.9937, -1.9937],\n",
      "           [-1.9918, -1.9918, -1.9919,  ..., -1.9940, -1.9940, -1.9940],\n",
      "           [-1.9919, -1.9919, -1.9920,  ..., -1.9942, -1.9942, -1.9942],\n",
      "           ...,\n",
      "           [-1.9900, -1.9900, -1.9900,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           [-1.9900, -1.9900, -1.9900,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           [-1.9900, -1.9900, -1.9900,  ..., -1.9911, -1.9911, -1.9911]],\n",
      "\n",
      "          [[-1.9908, -1.9909, -1.9909,  ..., -1.9939, -1.9939, -1.9939],\n",
      "           [-1.9912, -1.9912, -1.9912,  ..., -1.9942, -1.9942, -1.9942],\n",
      "           [-1.9916, -1.9916, -1.9917,  ..., -1.9945, -1.9945, -1.9945],\n",
      "           ...,\n",
      "           [-1.9905, -1.9905, -1.9905,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           [-1.9905, -1.9905, -1.9905,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           [-1.9904, -1.9904, -1.9904,  ..., -1.9911, -1.9911, -1.9911]]],\n",
      "\n",
      "\n",
      "         [[[-1.9971, -1.9974, -1.9979,  ..., -1.9967, -1.9963, -1.9961],\n",
      "           [-1.9969, -1.9973, -1.9979,  ..., -1.9974, -1.9971, -1.9969],\n",
      "           [-1.9968, -1.9972, -1.9978,  ..., -1.9983, -1.9981, -1.9980],\n",
      "           ...,\n",
      "           [-1.9982, -1.9985, -1.9990,  ..., -1.9971, -1.9967, -1.9964],\n",
      "           [-1.9981, -1.9985, -1.9990,  ..., -1.9969, -1.9967, -1.9965],\n",
      "           [-1.9980, -1.9984, -1.9989,  ..., -1.9965, -1.9964, -1.9963]],\n",
      "\n",
      "          [[-1.9967, -1.9968, -1.9969,  ..., -1.9971, -1.9964, -1.9960],\n",
      "           [-1.9968, -1.9969, -1.9971,  ..., -1.9975, -1.9970, -1.9966],\n",
      "           [-1.9969, -1.9972, -1.9974,  ..., -1.9980, -1.9977, -1.9975],\n",
      "           ...,\n",
      "           [-1.9986, -1.9988, -1.9990,  ..., -1.9965, -1.9964, -1.9962],\n",
      "           [-1.9987, -1.9989, -1.9991,  ..., -1.9968, -1.9964, -1.9960],\n",
      "           [-1.9987, -1.9988, -1.9991,  ..., -1.9965, -1.9962, -1.9961]],\n",
      "\n",
      "          [[-1.9943, -1.9946, -1.9949,  ..., -1.9948, -1.9944, -1.9941],\n",
      "           [-1.9943, -1.9946, -1.9951,  ..., -1.9947, -1.9943, -1.9941],\n",
      "           [-1.9943, -1.9947, -1.9952,  ..., -1.9945, -1.9943, -1.9941],\n",
      "           ...,\n",
      "           [-1.9988, -1.9991, -1.9994,  ..., -1.9970, -1.9965, -1.9960],\n",
      "           [-1.9986, -1.9989, -1.9994,  ..., -1.9959, -1.9962, -1.9965],\n",
      "           [-1.9984, -1.9988, -1.9993,  ..., -1.9954, -1.9959, -1.9964]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9948, -1.9948, -1.9949,  ..., -1.9977, -1.9977, -1.9977],\n",
      "           [-1.9951, -1.9951, -1.9952,  ..., -1.9977, -1.9977, -1.9977],\n",
      "           [-1.9955, -1.9955, -1.9955,  ..., -1.9976, -1.9976, -1.9976],\n",
      "           ...,\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9957, -1.9955, -1.9954],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9957, -1.9955, -1.9954],\n",
      "           [-1.9953, -1.9953, -1.9953,  ..., -1.9956, -1.9955, -1.9954]],\n",
      "\n",
      "          [[-1.9947, -1.9948, -1.9948,  ..., -1.9956, -1.9956, -1.9956],\n",
      "           [-1.9951, -1.9952, -1.9952,  ..., -1.9961, -1.9961, -1.9961],\n",
      "           [-1.9956, -1.9957, -1.9957,  ..., -1.9968, -1.9968, -1.9968],\n",
      "           ...,\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9958, -1.9958, -1.9958],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9958, -1.9958, -1.9958],\n",
      "           [-1.9954, -1.9954, -1.9954,  ..., -1.9959, -1.9959, -1.9959]],\n",
      "\n",
      "          [[-1.9940, -1.9940, -1.9941,  ..., -1.9967, -1.9967, -1.9967],\n",
      "           [-1.9946, -1.9946, -1.9946,  ..., -1.9971, -1.9971, -1.9971],\n",
      "           [-1.9953, -1.9954, -1.9954,  ..., -1.9977, -1.9977, -1.9977],\n",
      "           ...,\n",
      "           [-1.9958, -1.9958, -1.9958,  ..., -1.9960, -1.9960, -1.9960],\n",
      "           [-1.9958, -1.9958, -1.9958,  ..., -1.9960, -1.9960, -1.9960],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9959, -1.9959, -1.9959]]]],\n",
      "\n",
      "\n",
      "\n",
      "        [[[[-1.9886, -1.9888, -1.9891,  ..., -1.9918, -1.9914, -1.9912],\n",
      "           [-1.9884, -1.9886, -1.9889,  ..., -1.9923, -1.9921, -1.9919],\n",
      "           [-1.9881, -1.9883, -1.9887,  ..., -1.9931, -1.9929, -1.9928],\n",
      "           ...,\n",
      "           [-1.9979, -1.9983, -1.9988,  ..., -1.9911, -1.9910, -1.9908],\n",
      "           [-1.9979, -1.9983, -1.9988,  ..., -1.9909, -1.9910, -1.9910],\n",
      "           [-1.9978, -1.9982, -1.9987,  ..., -1.9910, -1.9910, -1.9910]],\n",
      "\n",
      "          [[-1.9911, -1.9914, -1.9917,  ..., -1.9925, -1.9921, -1.9918],\n",
      "           [-1.9915, -1.9917, -1.9920,  ..., -1.9927, -1.9923, -1.9920],\n",
      "           [-1.9919, -1.9921, -1.9924,  ..., -1.9929, -1.9925, -1.9922],\n",
      "           ...,\n",
      "           [-1.9984, -1.9985, -1.9988,  ..., -1.9908, -1.9910, -1.9910],\n",
      "           [-1.9985, -1.9987, -1.9989,  ..., -1.9912, -1.9909, -1.9907],\n",
      "           [-1.9983, -1.9985, -1.9988,  ..., -1.9910, -1.9907, -1.9906]],\n",
      "\n",
      "          [[-1.9908, -1.9910, -1.9912,  ..., -1.9919, -1.9914, -1.9909],\n",
      "           [-1.9907, -1.9910, -1.9912,  ..., -1.9923, -1.9915, -1.9909],\n",
      "           [-1.9907, -1.9909, -1.9913,  ..., -1.9927, -1.9918, -1.9910],\n",
      "           ...,\n",
      "           [-1.9985, -1.9988, -1.9992,  ..., -1.9909, -1.9910, -1.9911],\n",
      "           [-1.9982, -1.9987, -1.9993,  ..., -1.9908, -1.9910, -1.9911],\n",
      "           [-1.9979, -1.9985, -1.9992,  ..., -1.9908, -1.9909, -1.9911]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9897, -1.9898, -1.9898,  ..., -1.9932, -1.9932, -1.9932],\n",
      "           [-1.9898, -1.9898, -1.9899,  ..., -1.9936, -1.9936, -1.9936],\n",
      "           [-1.9899, -1.9899, -1.9900,  ..., -1.9941, -1.9941, -1.9941],\n",
      "           ...,\n",
      "           [-1.9883, -1.9883, -1.9883,  ..., -1.9887, -1.9886, -1.9885],\n",
      "           [-1.9883, -1.9883, -1.9883,  ..., -1.9887, -1.9886, -1.9885],\n",
      "           [-1.9882, -1.9882, -1.9882,  ..., -1.9890, -1.9889, -1.9888]],\n",
      "\n",
      "          [[-1.9898, -1.9898, -1.9899,  ..., -1.9923, -1.9923, -1.9923],\n",
      "           [-1.9900, -1.9900, -1.9901,  ..., -1.9926, -1.9926, -1.9926],\n",
      "           [-1.9902, -1.9902, -1.9903,  ..., -1.9930, -1.9930, -1.9930],\n",
      "           ...,\n",
      "           [-1.9882, -1.9882, -1.9882,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9882, -1.9882, -1.9882,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9883, -1.9883, -1.9883,  ..., -1.9893, -1.9893, -1.9893]],\n",
      "\n",
      "          [[-1.9892, -1.9892, -1.9893,  ..., -1.9915, -1.9915, -1.9915],\n",
      "           [-1.9896, -1.9896, -1.9896,  ..., -1.9919, -1.9919, -1.9919],\n",
      "           [-1.9900, -1.9900, -1.9901,  ..., -1.9924, -1.9924, -1.9924],\n",
      "           ...,\n",
      "           [-1.9887, -1.9887, -1.9887,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9887, -1.9887, -1.9887,  ..., -1.9889, -1.9889, -1.9889],\n",
      "           [-1.9886, -1.9886, -1.9886,  ..., -1.9893, -1.9893, -1.9893]]],\n",
      "\n",
      "\n",
      "         [[[-1.9947, -1.9948, -1.9949,  ..., -1.9936, -1.9933, -1.9930],\n",
      "           [-1.9946, -1.9947, -1.9949,  ..., -1.9940, -1.9938, -1.9936],\n",
      "           [-1.9945, -1.9946, -1.9948,  ..., -1.9945, -1.9943, -1.9942],\n",
      "           ...,\n",
      "           [-1.9982, -1.9986, -1.9991,  ..., -1.9932, -1.9930, -1.9929],\n",
      "           [-1.9982, -1.9986, -1.9991,  ..., -1.9930, -1.9931, -1.9930],\n",
      "           [-1.9981, -1.9985, -1.9990,  ..., -1.9930, -1.9929, -1.9929]],\n",
      "\n",
      "          [[-1.9928, -1.9931, -1.9935,  ..., -1.9940, -1.9936, -1.9933],\n",
      "           [-1.9929, -1.9932, -1.9936,  ..., -1.9943, -1.9939, -1.9936],\n",
      "           [-1.9931, -1.9933, -1.9937,  ..., -1.9948, -1.9943, -1.9940],\n",
      "           ...,\n",
      "           [-1.9986, -1.9988, -1.9991,  ..., -1.9928, -1.9927, -1.9926],\n",
      "           [-1.9988, -1.9989, -1.9991,  ..., -1.9931, -1.9926, -1.9922],\n",
      "           [-1.9987, -1.9988, -1.9991,  ..., -1.9927, -1.9923, -1.9921]],\n",
      "\n",
      "          [[-1.9917, -1.9920, -1.9922,  ..., -1.9930, -1.9925, -1.9921],\n",
      "           [-1.9918, -1.9921, -1.9924,  ..., -1.9931, -1.9925, -1.9920],\n",
      "           [-1.9920, -1.9922, -1.9925,  ..., -1.9931, -1.9925, -1.9919],\n",
      "           ...,\n",
      "           [-1.9990, -1.9992, -1.9994,  ..., -1.9930, -1.9927, -1.9926],\n",
      "           [-1.9987, -1.9991, -1.9995,  ..., -1.9930, -1.9929, -1.9928],\n",
      "           [-1.9987, -1.9990, -1.9994,  ..., -1.9929, -1.9929, -1.9929]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9916, -1.9916, -1.9917,  ..., -1.9943, -1.9943, -1.9943],\n",
      "           [-1.9916, -1.9916, -1.9917,  ..., -1.9948, -1.9948, -1.9948],\n",
      "           [-1.9916, -1.9916, -1.9917,  ..., -1.9954, -1.9954, -1.9954],\n",
      "           ...,\n",
      "           [-1.9900, -1.9900, -1.9900,  ..., -1.9905, -1.9904, -1.9903],\n",
      "           [-1.9900, -1.9900, -1.9900,  ..., -1.9905, -1.9904, -1.9903],\n",
      "           [-1.9899, -1.9899, -1.9899,  ..., -1.9909, -1.9907, -1.9906]],\n",
      "\n",
      "          [[-1.9917, -1.9917, -1.9917,  ..., -1.9937, -1.9937, -1.9937],\n",
      "           [-1.9918, -1.9918, -1.9919,  ..., -1.9940, -1.9940, -1.9940],\n",
      "           [-1.9919, -1.9919, -1.9920,  ..., -1.9942, -1.9942, -1.9942],\n",
      "           ...,\n",
      "           [-1.9900, -1.9900, -1.9900,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           [-1.9900, -1.9900, -1.9900,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           [-1.9900, -1.9900, -1.9900,  ..., -1.9911, -1.9911, -1.9911]],\n",
      "\n",
      "          [[-1.9908, -1.9909, -1.9909,  ..., -1.9939, -1.9939, -1.9939],\n",
      "           [-1.9912, -1.9912, -1.9912,  ..., -1.9942, -1.9942, -1.9942],\n",
      "           [-1.9916, -1.9916, -1.9917,  ..., -1.9945, -1.9945, -1.9945],\n",
      "           ...,\n",
      "           [-1.9905, -1.9905, -1.9905,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           [-1.9905, -1.9905, -1.9905,  ..., -1.9907, -1.9907, -1.9907],\n",
      "           [-1.9904, -1.9904, -1.9904,  ..., -1.9911, -1.9911, -1.9911]]],\n",
      "\n",
      "\n",
      "         [[[-1.9971, -1.9974, -1.9979,  ..., -1.9967, -1.9963, -1.9961],\n",
      "           [-1.9969, -1.9973, -1.9979,  ..., -1.9974, -1.9971, -1.9969],\n",
      "           [-1.9968, -1.9972, -1.9978,  ..., -1.9983, -1.9981, -1.9980],\n",
      "           ...,\n",
      "           [-1.9982, -1.9985, -1.9990,  ..., -1.9971, -1.9967, -1.9964],\n",
      "           [-1.9981, -1.9985, -1.9990,  ..., -1.9969, -1.9967, -1.9965],\n",
      "           [-1.9980, -1.9984, -1.9989,  ..., -1.9965, -1.9964, -1.9963]],\n",
      "\n",
      "          [[-1.9967, -1.9968, -1.9969,  ..., -1.9971, -1.9964, -1.9960],\n",
      "           [-1.9968, -1.9969, -1.9971,  ..., -1.9975, -1.9970, -1.9966],\n",
      "           [-1.9969, -1.9972, -1.9974,  ..., -1.9980, -1.9977, -1.9975],\n",
      "           ...,\n",
      "           [-1.9986, -1.9988, -1.9990,  ..., -1.9965, -1.9964, -1.9962],\n",
      "           [-1.9987, -1.9989, -1.9991,  ..., -1.9968, -1.9964, -1.9960],\n",
      "           [-1.9987, -1.9988, -1.9991,  ..., -1.9965, -1.9962, -1.9961]],\n",
      "\n",
      "          [[-1.9943, -1.9946, -1.9949,  ..., -1.9948, -1.9944, -1.9941],\n",
      "           [-1.9943, -1.9946, -1.9951,  ..., -1.9947, -1.9943, -1.9941],\n",
      "           [-1.9943, -1.9947, -1.9952,  ..., -1.9945, -1.9943, -1.9941],\n",
      "           ...,\n",
      "           [-1.9988, -1.9991, -1.9994,  ..., -1.9970, -1.9965, -1.9960],\n",
      "           [-1.9986, -1.9989, -1.9994,  ..., -1.9959, -1.9962, -1.9965],\n",
      "           [-1.9984, -1.9988, -1.9993,  ..., -1.9954, -1.9959, -1.9964]],\n",
      "\n",
      "          ...,\n",
      "\n",
      "          [[-1.9948, -1.9948, -1.9949,  ..., -1.9977, -1.9977, -1.9977],\n",
      "           [-1.9951, -1.9951, -1.9952,  ..., -1.9977, -1.9977, -1.9977],\n",
      "           [-1.9955, -1.9955, -1.9955,  ..., -1.9976, -1.9976, -1.9976],\n",
      "           ...,\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9957, -1.9955, -1.9954],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9957, -1.9955, -1.9954],\n",
      "           [-1.9953, -1.9953, -1.9953,  ..., -1.9956, -1.9955, -1.9954]],\n",
      "\n",
      "          [[-1.9947, -1.9948, -1.9948,  ..., -1.9956, -1.9956, -1.9956],\n",
      "           [-1.9951, -1.9952, -1.9952,  ..., -1.9961, -1.9961, -1.9961],\n",
      "           [-1.9956, -1.9957, -1.9957,  ..., -1.9968, -1.9968, -1.9968],\n",
      "           ...,\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9958, -1.9958, -1.9958],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9958, -1.9958, -1.9958],\n",
      "           [-1.9954, -1.9954, -1.9954,  ..., -1.9959, -1.9959, -1.9959]],\n",
      "\n",
      "          [[-1.9940, -1.9940, -1.9941,  ..., -1.9967, -1.9967, -1.9967],\n",
      "           [-1.9946, -1.9946, -1.9946,  ..., -1.9971, -1.9971, -1.9971],\n",
      "           [-1.9953, -1.9954, -1.9954,  ..., -1.9977, -1.9977, -1.9977],\n",
      "           ...,\n",
      "           [-1.9958, -1.9958, -1.9958,  ..., -1.9960, -1.9960, -1.9960],\n",
      "           [-1.9958, -1.9958, -1.9958,  ..., -1.9960, -1.9960, -1.9960],\n",
      "           [-1.9956, -1.9956, -1.9956,  ..., -1.9959, -1.9959, -1.9959]]]]])]\n",
      "question:  ('What does the woman do before places hands in lap ?', 'What does the woman do after has hands on face ?', 'What does the woman do before pushes top towards herself ?', 'What does the woman do after leaning down ?', 'What does the woman do before turn to the right side ?', 'What does the woman do after look left side ?', 'What does the man on left do before fall down ?', 'What does the man on left do after get pushed by man ?', 'What does the man do before raises arms to cover face ?', 'What does the man do after laugh ?', 'What does the woman do before turn head to the left ?', 'What does the woman do after fold a spoon ?', 'What does the woman do before turn away ?', 'What does the woman do after sip from cup ?', 'What does the man do before move back ?', 'What does the man do after look stationary ?', 'What does the metal spoon do before bend completely by girl ?', 'What does the metal spoon do after in girls hand ?', 'What does the girl do before begin to dance vigorously ?', 'What does the girl do after strike a pose ?', 'What does the two men do before lower their head ?', 'What does the two men do after look bore ?', 'What does the boy do before look up ?', 'What does the boy do after look down ?', 'What does the girl do before look worry ?', 'What does the girl do after smile ?', 'What does the man in sweater do before scream with anger ?', 'What does the man in sweater do after lay on bed ?', 'What does the man in black suit do before sit on floor ?', 'What does the man in black suit do after stand ?', 'What does the woman do before raise arm up ?', 'What does the woman do after sway hips and arm ?')\n",
      "options:  [(\"stoke man's face\", 'vault over wall', 'leaning down', 'thrust hip', 'pout lower lips', 'turn to the right side', 'get pushed by man', 'yell', 'surf up wave', 'oust the girl who stand on the right', 'fold a spoon', 'strum guitar', 'calm down', 'cover face with hand', 'look stationary', 'pet the critter', 'summon lightning', 'wink', 'syrup continue to pour down', 'cover face', 'withdraw an ice cube', 'hop', 'frown', 'look up', 'growl', 'clap another guy hand', 'sway hips', 'itch head', 'await a response', 'clasp hand', 'look scared', 'twirl girl around'), ('wink', \"stoke man's face\", 'disgust', 'sing', 'dive', 'dance', 'wave arm', 'multiply card in hand', 'snuggle man', 'paw down', 'elevate legs into the air', 'turn head to the left', 'look tensed', 'strum the guitar', 'pucker lips', 'tilt head', 'bow back', 'bend completely by girl', 'strike a pose', 'zip out lips', 'outstretch and raise hand', 'hug dog', \"film the woman's torso\", 'pat on knee', 'mime as if throw something', 'outstretch hand', 'suspicious', 'clasp hand together', 'blink eye', 'sit on floor', 'skeptical', 'raise arm up'), ('chew food', 'places hands in lap', 'squint', 'pushes top towards herself', 'scoot forward', \"stoke man's face\", 'dance in the chair', 'itch head', 'skid across the pavement', 'wipe away tear', 'smoke', 'dance with another boy', 'kiss man', 'whistle', 'outstretch arms', 'cover face with hand', 'clap hand', 'twirl around', 'smile', 'swivel shoulder around', 'wave tongue', 'uncover face', 'look down', 'outstretch arm away from body', 'smile', 'holds cigarette', 'dance to the left', 'scream with anger', 'giggle', 'sniff something', 'crawl', 'furious'), ('clap the steer wheel', 'lean in and kiss her', 'remove jacket', 'watch', 'wave hand', 'bow over', \"stoke man's face\", 'fall down', 'tilt head', 'raises arms to cover face', 'grimace', 'whistle', 'sip from cup', 'hula hoop', 'withdraw an ice cube', 'stalk', 'in girls hand', 'crash into wall', 'enjoys massage', 'begin to dance vigorously', 'wink', 'lower their head', 'perform backflip', 'grip cabinet', 'wink eye', 'smirk', 'maneuver foreward', 'fuss with sail', 'arch backwards', 'nibble pancake', 'angry', \"thump towrad woman's side\"), ('has hands on face', 'slash club', \"stoke man's face\", \"stoke man's face\", 'look left side', 'descend from a platform', 'twirl', 'fuss with sail', 'laugh', 'widen arm', 'run left', 'lick cigar', 'skeptical', 'turn away', 'wave', 'move back', 'scream', 'talks', 'swipe hair', 'bar teeth', 'look bore', 'squint eye', 'outstretch arm away from body', 'inhale through nose', 'adjust glass', 'look worry', 'lay on bed', \"paw at dog's paw\", 'stand', 'flex', 'sway hips and arm', 'yawn')]\n",
      "answer:  ('4', '2', '0', '2', '4', '0', '0', '3', '4', '3', '0', '1', '3', '4', '0', '4', '3', '1', '1', '3', '4', '3', '2', '0', '2', '4', '4', '2', '4', '1', '4', '1')\n"
     ]
    }
   ],
   "source": [
    "dataloader = DataLoader(dataset, batch_size=32)\n",
    "for tensor, question, options, answer in dataloader:\n",
    "    print(\"tensor: \", tensor)\n",
    "    print(\"question: \", question)\n",
    "    print(\"options: \", options)\n",
    "    print(\"answer: \", answer)\n",
    "    break"
   ]
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  },
  {
   "cell_type": "code",
   "execution_count": null,
   "metadata": {},
   "outputs": [],
   "source": []
  }
 ],
 "metadata": {
  "kernelspec": {
   "display_name": "Python 3",
   "language": "python",
   "name": "python3"
  },
  "language_info": {
   "codemirror_mode": {
    "name": "ipython",
    "version": 3
   },
   "file_extension": ".py",
   "mimetype": "text/x-python",
   "name": "python",
   "nbconvert_exporter": "python",
   "pygments_lexer": "ipython3",
   "version": "3.6.9"
  }
 },
 "nbformat": 4,
 "nbformat_minor": 2
}
